[BugFix] Fix CPU inference failure (#15231)
This commit is contained in:
@@ -1443,7 +1443,9 @@ class ServerArgs:
|
|||||||
# TODO: currently, it is only supported in the single node scenario. https://github.com/flashinfer-ai/flashinfer/issues/2006
|
# TODO: currently, it is only supported in the single node scenario. https://github.com/flashinfer-ai/flashinfer/issues/2006
|
||||||
# TODO: there is currently a bug on H20 device specifically, https://github.com/flashinfer-ai/flashinfer/issues/2204
|
# TODO: there is currently a bug on H20 device specifically, https://github.com/flashinfer-ai/flashinfer/issues/2204
|
||||||
device_name = get_device_name()
|
device_name = get_device_name()
|
||||||
is_h20_device = "H20" in device_name and "H200" not in device_name
|
is_h20_device = (
|
||||||
|
device_name and "H20" in device_name and "H200" not in device_name
|
||||||
|
)
|
||||||
if (
|
if (
|
||||||
not self.enable_flashinfer_allreduce_fusion
|
not self.enable_flashinfer_allreduce_fusion
|
||||||
and model_arch
|
and model_arch
|
||||||
|
|||||||
Reference in New Issue
Block a user