Skip to content

Commit a14d6f6

Browse files
authored
Merge pull request #202 from lcskrishna/cl/enable-fp16-nn
enabled fp16 tests that work with compiler fixes in test_nn
2 parents d20a8e4 + dbf77c8 commit a14d6f6

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

test/test_nn.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6556,7 +6556,6 @@ def add(test_name, fn):
65566556
add(cuda_test_name + '_double', lambda self,
65576557
test=test, kwargs=kwargs: test.test_cuda(self, dtype=torch.double, **kwargs))
65586558

6559-
@skipIfRocm
65606559
def test_half(self, test=test, kwargs=kwargs):
65616560
test.test_cuda(self, dtype=torch.half, **kwargs)
65626561
if getattr(test, 'check_half', True):

0 commit comments

Comments
 (0)