We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 44b32df commit 24cd909Copy full SHA for 24cd909
test/regressions/test_copy.py
@@ -12,7 +12,9 @@
12
13
14
class TestSimpleCopy(TestCase):
15
- @dtypes(*float8_types_and(torch.float8_e8m0fnu, torch.float32))
+ @dtypes(
16
+ *float8_types_and(torch.float8_e8m0fnu, torch.float4_e2m1fn_x2, torch.float32)
17
+ )
18
def test_copy_and_clone(self, dtype):
19
a_cpu = torch.randn(16, 64, 28, 28)
20
b_cpu = torch.randn(16, 64, 28, 28)
0 commit comments