summaryrefslogtreecommitdiff
path: root/csrc
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-08-23 16:00:26 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-08-23 16:00:26 -0700
commitee5b947e63c2340405f25e4e83066f39292bc0ed (patch)
tree70fa3b2041e1cedf257feeff4e5b071703c43c2a /csrc
parent7e0fb655e1e040221054886fbee9d5682aa6e4e2 (diff)
Fixed issue where Pascal was not displaying proper error.
Diffstat (limited to 'csrc')
-rw-r--r--csrc/ops.cu6
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;