diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
| commit | 8b1fd32e3e4f5073fd055cb5f9261ec585f8cc2c (patch) | |
| tree | 76044424d73d02e1026c996b22b9da5061188387 /tests | |
| parent | 7d2ecd30c044840ba5f161ec73e5eaf30ac8131d (diff) | |
Fixed makefile; fixed Ampere igemmlt_8 bug.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_functional.py | 1 |
1 files changed, 1 insertions, 0 deletions
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 ) |