diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-16 19:03:19 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-16 19:03:19 -0700 |
| commit | a6664de0720c7d8572a475a9c59f7dd85b5f83b0 (patch) | |
| tree | 1342a757dfc60fe37f1d94371251fd17bc968bff /csrc/ops.cu | |
| parent | de354f7ded52bfa857089769225cdf1ee694bfd6 (diff) | |
Enhanced error handling in CUDA SETUP failures.
Diffstat (limited to 'csrc/ops.cu')
| -rw-r--r-- | csrc/ops.cu | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/csrc/ops.cu b/csrc/ops.cu index ed32828..c0ec3cb 100644 --- a/csrc/ops.cu +++ b/csrc/ops.cu @@ -371,6 +371,9 @@ template void transform<int32_t, COL32, ROW, false, 32>(cublasLtHandle_t ltHandl template <int FORMATB, int DTYPE_OUT, int SCALE_ROWS> int igemmlt(cublasLtHandle_t ltHandle, int m, int n, int k, const int8_t *A, const int8_t *B, void *C, float *row_scale, int lda, int ldb, int ldc) { #ifdef NO_CUBLASLT + printf("ERROR: Your GPU does not support Int8 Matmul!"); + assert(false); + return 0; #else int has_error = 0; |