add bf16 override for directml

Signed-off-by: Vladimir Mandic <[email protected]>
vladmandic · Sep 30, 2024 · c21a10b · c21a10b
1 parent d0f9450
commit c21a10b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/devices.py b/modules/devices.py
@@ -272,7 +272,7 @@ def test_bf16():
     global bf16_ok # pylint: disable=global-statement
     if bf16_ok is not None:
         return bf16_ok
-    if sys.platform == "darwin" or backend == 'openvino': # override
+    if sys.platform == "darwin" or backend == 'openvino' or backend == 'directml': # override
         bf16_ok = False
         return bf16_ok
     try: