Skip to content

Commit fe067e7

Browse files
committed
test/test_transformers: sanity check of golden tensor
1 parent 0e4f9a3 commit fe067e7

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

test/test_transformers.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,12 @@ def _check_equal(
129129
_check_equal(gold, ref, tst, fudge_factor, tensor_name)
130130
return
131131

132+
if golden.is_cuda and golden.dtype == torch.float32:
133+
assert torch.backends.cuda.math_sdp.fp32_precision == "ieee", (
134+
"Testing script error: FP32 golden tensor must be calculated with IEEE"
135+
" precision"
136+
)
137+
132138
# Compute error between golden
133139
test_error = (golden - test).abs().max()
134140
ref_error = (golden - reference).abs().max()

0 commit comments

Comments
 (0)