This commit is contained in:
Disty0
2024-10-24 20:08:49 +03:00
parent 3acf6013f6
commit d459acfcca
2 changed files with 2 additions and 1 deletions
+1 -1
View File
@@ -285,7 +285,7 @@ def test_bf16():
return bf16_ok
elif backend == 'zluda':
device_name = torch.cuda.get_device_name(device)
if "AMD Radeon RX " in device_name: # only force AMD
if device_name.startswith("AMD Radeon RX "): # only force AMD
device_name = device_name.replace("AMD Radeon RX ", "").split(" ", maxsplit=1)[0]
if len(device_name) == 4 and device_name[0] in {"5", "6"}: # RDNA 1 and 2
bf16_ok = False
+1
View File
@@ -296,6 +296,7 @@ def openvino_compile_cached_model(cached_model_path, *example_inputs):
hints[ov_hints.execution_mode] = ov_hints.ExecutionMode.PERFORMANCE
elif shared.opts.openvino_accuracy == "accuracy":
hints[ov_hints.execution_mode] = ov_hints.ExecutionMode.ACCURACY
core.set_property(hints)
dont_use_nncf = False
dont_use_quant = False
dont_use_4bit_nncf = False