From 8b1fd32e3e4f5073fd055cb5f9261ec585f8cc2c Mon Sep 17 00:00:00 2001
From: Tim Dettmers <tim.dettmers@gmail.com>
Date: Mon, 25 Jul 2022 14:02:14 -0700
Subject: Fixed makefile; fixed Ampere igemmlt_8 bug.

---
 tests/test_functional.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'tests')

diff --git a/tests/test_functional.py b/tests/test_functional.py
index 6cbe58f..dcb0255 100644
--- a/tests/test_functional.py
+++ b/tests/test_functional.py
@@ -1183,6 +1183,7 @@ def test_transform_to_row(dim1, dim2, dtype, orderA, orderOut):
 
 def test_overflow():
     formatB = F.get_special_format_str()
+    print(formatB)
     for i in range(2):
         a = torch.arange(5, 15).cuda().to(torch.int8).view(-1,1 )
         b = torch.arange(5, 15).cuda().to(torch.int8).view(-1,1 )
-- 
cgit v1.2.3