diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-09-05 16:23:03 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-09-05 16:23:03 -0700 |
| commit | eab4d8232d558f2e6bd7f7cc3d00e2e6e94f4e80 (patch) | |
| tree | 1a70991e0d9d5632a20388bf87e4c56dfb06d64e /csrc | |
| parent | 5d9bc7a301618d6642a70951ac46637d4889c82b (diff) | |
| parent | 8e7053a988f7bf6c0b2605a3d0ec461f5d9b05e4 (diff) | |
Merge branch 'bugfix_cpu_and_cc' into main
Diffstat (limited to 'csrc')
| -rw-r--r-- | csrc/ops.cu | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/csrc/ops.cu b/csrc/ops.cu index c0ec3cb..e49c94b 100644 --- a/csrc/ops.cu +++ b/csrc/ops.cu @@ -371,7 +371,11 @@ 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!"); + cout << "" << endl; + cout << "=============================================" << endl; + cout << "ERROR: Your GPU does not support Int8 Matmul!" << endl; + cout << "=============================================" << endl; + cout << "" << endl; assert(false); return 0; |