Skip to content

Commit 8b775aa

Browse files
committed
Temporarily comment out create_hal_driver usage for old iree version compat (DNM)
1 parent 4692e11 commit 8b775aa

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

models/turbine_models/model_runner.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import argparse
22
import sys
33
from iree import runtime as ireert
4-
from iree.runtime._binding import create_hal_driver
4+
#from iree.runtime._binding import create_hal_driver
55

66

77
class vmfbRunner:
@@ -11,14 +11,14 @@ def __init__(self, device, vmfb_path, external_weight_path=None, extra_plugin=No
1111
# If an extra plugin is requested, add a global flag to load the plugin
1212
# and create the driver using the non-caching creation function, as
1313
# the caching creation function may ignore the flag.
14-
if extra_plugin:
15-
ireert.flags.parse_flags(f"--executable_plugin={extra_plugin}")
16-
haldriver = create_hal_driver(device)
14+
# if extra_plugin:
15+
# ireert.flags.parse_flags(f"--executable_plugin={extra_plugin}")
16+
# haldriver = create_hal_driver(device)
1717

1818
# No plugin requested: create the driver with the caching create
1919
# function.
20-
else:
21-
haldriver = ireert.get_driver(device)
20+
#else:
21+
haldriver = ireert.get_driver(device)
2222
if "://" in device:
2323
try:
2424
device_idx = int(device.split("://")[-1])

0 commit comments

Comments
 (0)