Unverified Commit dafb4e50 authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[V1][Bugfix] Fix oracle for device checking (#15104)


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent 68cf1601
...@@ -1578,6 +1578,13 @@ class EngineArgs: ...@@ -1578,6 +1578,13 @@ class EngineArgs:
_raise_or_fallback(feature_name=name, recommend_to_remove=True) _raise_or_fallback(feature_name=name, recommend_to_remove=True)
return False return False
# No support for device type other than CUDA, AMD (experiemntal) or
# TPU (experimental) so far.
if not (current_platform.is_cuda_alike() or current_platform.is_tpu()):
_raise_or_fallback(
feature_name=f"device type={current_platform.device_type}",
recommend_to_remove=False)
return False
############################################################# #############################################################
# Experimental Features - allow users to opt in. # Experimental Features - allow users to opt in.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment