Skip to content

Commit 24cd909

Browse files
committed
Add test case
1 parent 44b32df commit 24cd909

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/regressions/test_copy.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@
1212

1313

1414
class TestSimpleCopy(TestCase):
15-
@dtypes(*float8_types_and(torch.float8_e8m0fnu, torch.float32))
15+
@dtypes(
16+
*float8_types_and(torch.float8_e8m0fnu, torch.float4_e2m1fn_x2, torch.float32)
17+
)
1618
def test_copy_and_clone(self, dtype):
1719
a_cpu = torch.randn(16, 64, 28, 28)
1820
b_cpu = torch.randn(16, 64, 28, 28)

0 commit comments

Comments
 (0)