Skip to content

Commit 6e70f68

Browse files
author
Aidan
committed
Disable int8, gemm_batch for cublas
1 parent 7f72a3d commit 6e70f68

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/blas/backends/cublas/cublas_batch.cpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,6 @@ inline void gemm_batch_impl(sycl::queue &queue, transpose transa, transpose tran
190190

191191
GEMM_STRIDED_BATCH_LAUNCHER(sycl::half, sycl::half, sycl::half, sycl::half)
192192
GEMM_STRIDED_BATCH_LAUNCHER(sycl::half, sycl::half, float, float)
193-
GEMM_STRIDED_BATCH_LAUNCHER(std::int8_t, std::int8_t, float, float)
194193
GEMM_STRIDED_BATCH_LAUNCHER(float, float, float, float)
195194
GEMM_STRIDED_BATCH_LAUNCHER(double, double, double, double)
196195
GEMM_STRIDED_BATCH_LAUNCHER(std::complex<float>, std::complex<float>, std::complex<float>,
@@ -212,6 +211,7 @@ GEMM_STRIDED_BATCH_LAUNCHER(std::complex<double>, std::complex<double>, std::com
212211
dtype_string<TYPE_C>() + "," + dtype_string<TYPE_S>() + ">"); \
213212
}
214213

214+
GEMM_STRIDED_BATCH_LAUNCHER(std::int8_t, std::int8_t, float, float)
215215
GEMM_STRIDED_BATCH_LAUNCHER(std::int8_t, std::int8_t, std::int32_t, float)
216216

217217
#undef GEMM_STRIDED_BATCH_LAUNCHER
@@ -632,7 +632,6 @@ inline sycl::event gemm_batch_strided_usm_impl(sycl::queue &queue, transpose tra
632632

633633
GEMM_STRIDED_BATCH_LAUNCHER_USM(sycl::half, sycl::half, sycl::half, sycl::half)
634634
GEMM_STRIDED_BATCH_LAUNCHER_USM(sycl::half, sycl::half, float, float)
635-
GEMM_STRIDED_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, float, float)
636635
GEMM_STRIDED_BATCH_LAUNCHER_USM(float, float, float, float)
637636
GEMM_STRIDED_BATCH_LAUNCHER_USM(double, double, double, double)
638637
GEMM_STRIDED_BATCH_LAUNCHER_USM(std::complex<float>, std::complex<float>, std::complex<float>,
@@ -654,6 +653,7 @@ GEMM_STRIDED_BATCH_LAUNCHER_USM(std::complex<double>, std::complex<double>, std:
654653
dtype_string<TYPE_C>() + "," + dtype_string<TYPE_S>() + ">"); \
655654
}
656655

656+
GEMM_STRIDED_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, float, float)
657657
GEMM_STRIDED_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, std::int32_t, float)
658658

659659
#undef GEMM_STRIDED_BATCH_LAUNCHER_USM
@@ -714,7 +714,6 @@ inline sycl::event gemm_batch_usm_impl(sycl::queue &queue, transpose *transa, tr
714714

715715
GEMM_BATCH_LAUNCHER_USM(sycl::half, sycl::half, sycl::half, sycl::half)
716716
GEMM_BATCH_LAUNCHER_USM(sycl::half, sycl::half, float, float)
717-
GEMM_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, float, float)
718717
GEMM_BATCH_LAUNCHER_USM(float, float, float, float)
719718
GEMM_BATCH_LAUNCHER_USM(double, double, double, double)
720719
GEMM_BATCH_LAUNCHER_USM(std::complex<float>, std::complex<float>, std::complex<float>,
@@ -736,6 +735,7 @@ GEMM_BATCH_LAUNCHER_USM(std::complex<double>, std::complex<double>, std::complex
736735
dtype_string<TYPE_C>() + "," + dtype_string<TYPE_S>() + ">"); \
737736
}
738737

738+
GEMM_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, float, float)
739739
GEMM_BATCH_LAUNCHER_USM(std::int8_t, std::int8_t, std::int32_t, float)
740740

741741
#undef GEMM_BATCH_LAUNCHER_USM

0 commit comments

Comments
 (0)