From a6664de0720c7d8572a475a9c59f7dd85b5f83b0 Mon Sep 17 00:00:00 2001
From: Tim Dettmers <tim.dettmers@gmail.com>
Date: Tue, 16 Aug 2022 19:03:19 -0700
Subject: Enhanced error handling in CUDA SETUP failures.

---
 csrc/ops.cu | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'csrc')

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;
-- 
cgit v1.2.3