Skip to content

Commit 82605a9

Browse files
yasahi-hpcYuuichi Asahi
yasahi-hpc
and
Yuuichi Asahi
authored
batched/dense/impl/KokkosBatched_Gemm_Team_Impl.hpp (kokkos#2626)
Signed-off-by: Yuuichi Asahi <[email protected]> Co-authored-by: Yuuichi Asahi <[email protected]>
1 parent d3404ae commit 82605a9

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

‎batched/dense/impl/KokkosBatched_Gemm_Team_Impl.hpp‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ struct TeamGemm<MemberType, Trans::NoTranspose, Trans::NoTranspose, ArgAlgo> {
4848

4949
// Quick return if possible
5050
const int m = C_extent_0, n = C_extent_1, k = A_extent_1;
51-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
51+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
5252

5353
auto info = Impl::checkGemmInput<Trans::NoTranspose, Trans::NoTranspose>(A, B, C);
5454
if (info) return info;
@@ -79,7 +79,7 @@ struct TeamGemm<MemberType, Trans::Transpose, Trans::NoTranspose, ArgAlgo> {
7979

8080
// Quick return if possible
8181
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
82-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
82+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
8383

8484
auto info = Impl::checkGemmInput<Trans::Transpose, Trans::NoTranspose>(A, B, C);
8585
if (info) return info;
@@ -110,7 +110,7 @@ struct TeamGemm<MemberType, Trans::ConjTranspose, Trans::NoTranspose, ArgAlgo> {
110110

111111
// Quick return if possible
112112
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
113-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
113+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
114114

115115
auto info = Impl::checkGemmInput<Trans::ConjTranspose, Trans::NoTranspose>(A, B, C);
116116
if (info) return info;
@@ -141,7 +141,7 @@ struct TeamGemm<MemberType, Trans::NoTranspose, Trans::Transpose, ArgAlgo> {
141141

142142
// Quick return if possible
143143
const int m = C_extent_0, n = C_extent_1, k = A_extent_1;
144-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
144+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
145145

146146
auto info = Impl::checkGemmInput<Trans::NoTranspose, Trans::Transpose>(A, B, C);
147147
if (info) return info;
@@ -172,7 +172,7 @@ struct TeamGemm<MemberType, Trans::Transpose, Trans::Transpose, ArgAlgo> {
172172

173173
// Quick return if possible
174174
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
175-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
175+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
176176

177177
auto info = Impl::checkGemmInput<Trans::Transpose, Trans::Transpose>(A, B, C);
178178
if (info) return info;
@@ -203,7 +203,7 @@ struct TeamGemm<MemberType, Trans::ConjTranspose, Trans::Transpose, ArgAlgo> {
203203

204204
// Quick return if possible
205205
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
206-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
206+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
207207

208208
auto info = Impl::checkGemmInput<Trans::ConjTranspose, Trans::Transpose>(A, B, C);
209209
if (info) return info;
@@ -234,7 +234,7 @@ struct TeamGemm<MemberType, Trans::NoTranspose, Trans::ConjTranspose, ArgAlgo> {
234234

235235
// Quick return if possible
236236
const int m = C_extent_0, n = C_extent_1, k = A_extent_1;
237-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
237+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
238238

239239
auto info = Impl::checkGemmInput<Trans::NoTranspose, Trans::ConjTranspose>(A, B, C);
240240
if (info) return info;
@@ -265,7 +265,7 @@ struct TeamGemm<MemberType, Trans::Transpose, Trans::ConjTranspose, ArgAlgo> {
265265

266266
// Quick return if possible
267267
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
268-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
268+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
269269

270270
auto info = Impl::checkGemmInput<Trans::Transpose, Trans::ConjTranspose>(A, B, C);
271271
if (info) return info;
@@ -296,7 +296,7 @@ struct TeamGemm<MemberType, Trans::ConjTranspose, Trans::ConjTranspose, ArgAlgo>
296296

297297
// Quick return if possible
298298
const int m = C_extent_0, n = C_extent_1, k = A_extent_0;
299-
if (m == 0 || n == 0 || ((alpha == ScalarType(0) || k == 0) && beta == ScalarType(1))) return 0;
299+
if (m == 0 || n == 0 || k == 0 || (alpha == ScalarType(0) && beta == ScalarType(1))) return 0;
300300

301301
auto info = Impl::checkGemmInput<Trans::ConjTranspose, Trans::ConjTranspose>(A, B, C);
302302
if (info) return info;

0 commit comments

Comments
 (0)