Skip to content

Commit

Permalink
update cond
Browse files Browse the repository at this point in the history
Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
  • Loading branch information
crcrpar committed Dec 19, 2024
1 parent d4c28e3 commit 7fb186f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion thunder/tests/test_tensor_subclass.py
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ def test_torchao_float8_linear(executor, device, dtype, bias):
expected = fp8_model(x)
jitted = executor.make_callable(fp8_model)

if bias and dtype == thunder.core.dtypes.bfloat16 and executor == nvFuserExecutor:
if dtype == thunder.core.dtypes.bfloat16 and executor == nvFuserExecutor:
with pytest.raises(
RuntimeError, match="Failed to compute the min-cut on the graph due to a path with infinite capacity"
):
Expand Down

0 comments on commit 7fb186f

Please sign in to comment.