We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
is_compile_supported()
device_type
1 parent 49c3f18 commit d71863dCopy full SHA for d71863d
userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -3751,10 +3751,11 @@ def build_checkpoint_variable(**options):
3751
def is_compile_supported(device_type):
3752
from .eval_frame import is_dynamo_supported
3753
3754
+ type = torch.device(device_type).type
3755
compile_supported = is_dynamo_supported()
- if device_type == "cpu":
3756
+ if type == "cpu":
3757
pass
- elif device_type in ["cuda", "xpu"] and compile_supported:
3758
+ elif type in ["cuda", "xpu"] and compile_supported:
3759
compile_supported = has_triton()
3760
else:
3761
compile_supported = False
0 commit comments