Skip to content

Commit c4d5a2b

Browse files
committed
skip tests for UNetRLModelTests; needs next pytorch release for currently unimplemented operation support
1 parent a9364bd commit c4d5a2b

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

tests/models/unets/test_models_unet_1d.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,3 +292,21 @@ def test_output_pretrained(self):
292292
def test_forward_with_norm_groups(self):
293293
# Not implemented yet for this UNet
294294
pass
295+
296+
@unittest.skip(
297+
"RuntimeError: 'fill_out' not implemented for 'Float8_e4m3fn'. The error is caused due to certain torch.float8_e4m3fn and torch.float8_e5m2 operations "
298+
"not being supported when using deterministic algorithms (which is what the tests run with). To fix:\n"
299+
"1. Wait for next PyTorch release: https://github.com/pytorch/pytorch/issues/137160.\n"
300+
"2. Unskip this test."
301+
)
302+
def test_layerwise_casting_inference(self):
303+
pass
304+
305+
@unittest.skip(
306+
"RuntimeError: 'fill_out' not implemented for 'Float8_e4m3fn'. The error is caused due to certain torch.float8_e4m3fn and torch.float8_e5m2 operations "
307+
"not being supported when using deterministic algorithms (which is what the tests run with). To fix:\n"
308+
"1. Wait for next PyTorch release: https://github.com/pytorch/pytorch/issues/137160.\n"
309+
"2. Unskip this test."
310+
)
311+
def test_layerwise_casting_memory(self):
312+
pass

0 commit comments

Comments
 (0)