Skip to content

Commit 1d306b8

Browse files
committed
skip tests for UNet1DModelTests - unsupported pytorch operations
1 parent 0a16826 commit 1d306b8

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
@@ -152,6 +152,24 @@ def test_unet_1d_maestro(self):
152152
assert (output_sum - 224.0896).abs() < 0.5
153153
assert (output_max - 0.0607).abs() < 4e-4
154154

155+
@unittest.skip(
156+
"RuntimeError: 'fill_out' not implemented for 'Float8_e4m3fn'. The error is caused due to certain torch.float8_e4m3fn and torch.float8_e5m2 operations "
157+
"not being supported when using deterministic algorithms (which is what the tests run with). To fix:\n"
158+
"1. Wait for next PyTorch release: https://github.com/pytorch/pytorch/issues/137160.\n"
159+
"2. Unskip this test."
160+
)
161+
def test_layerwise_casting_inference(self):
162+
pass
163+
164+
@unittest.skip(
165+
"RuntimeError: 'fill_out' not implemented for 'Float8_e4m3fn'. The error is caused due to certain torch.float8_e4m3fn and torch.float8_e5m2 operations "
166+
"not being supported when using deterministic algorithms (which is what the tests run with). To fix:\n"
167+
"1. Wait for next PyTorch release: https://github.com/pytorch/pytorch/issues/137160.\n"
168+
"2. Unskip this test."
169+
)
170+
def test_layerwise_casting_memory(self):
171+
pass
172+
155173

156174
class UNetRLModelTests(ModelTesterMixin, UNetTesterMixin, unittest.TestCase):
157175
model_class = UNet1DModel

0 commit comments

Comments
 (0)