Skip to content

Commit 8efa4b6

Browse files
authored
[TEST] Fix for test nondeterminism on tl.dot-BF16xN input_precision (#8818)
We have had a number of test failures due to numpy_random. I suspect blowing the bottom bits as was already the case for TF32 is needed. Example test failures: https://github.com/triton-lang/triton/actions/runs/19592177401/job/56111902929#step:13:943 https://github.com/triton-lang/triton/actions/runs/19446507385/job/55645510809#step:13:946 https://github.com/triton-lang/triton/actions/runs/18967311789/job/54200126681?pr=8593#step:12:1000
1 parent 5dc3032 commit 8efa4b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/unit/language/test_core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3325,7 +3325,7 @@ def kernel(X, stride_xm, stride_xk, Y, stride_yk, stride_yn, W, stride_wn, strid
33253325
if 'int' not in in_dtype and 'float8' not in in_dtype:
33263326
x *= .1
33273327
y *= .1
3328-
if in_dtype == 'float32' and input_precision == "tf32":
3328+
if in_dtype == 'float32' and input_precision in ["tf32", "bf16x3", "bf16x6"]:
33293329
x = (x.view('uint32') & np.uint32(0xffffe000)).view('float32')
33303330
y = (y.view('uint32') & np.uint32(0xffffe000)).view('float32')
33313331
w = (w.view('uint32') & np.uint32(0xffffe000)).view('float32')

0 commit comments

Comments
 (0)