Merge pull request #373 from KernelTuner/fix-find_bfloat16_if_available

stijnh · web-flow · commit 04176b3828dd · 2026-04-07T13:00:11.000+02:00
Fix `_find_bfloat16_if_available` always returning `None`
diff --git a/kernel_tuner/accuracy.py b/kernel_tuner/accuracy.py
@@ -96,7 +96,7 @@ def _find_bfloat16_if_available():
             + "please install either the package `ml_dtypes`, `jax`, or `tensorflow`"
         )
 
-    return None
+    return dtype
 
 
 def _to_float_dtype(x: str) -> np.dtype:

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ def _find_bfloat16_if_available():`
`96`	`96`	+ "please install either the package `ml_dtypes`, `jax`, or `tensorflow`"
`97`	`97`	`)`
`98`	`98`
`99`		`- return None`
	`99`	`+ return dtype`
`100`	`100`
`101`	`101`
`102`	`102`	`def _to_float_dtype(x: str) -> np.dtype:`