Skip to content

GitLab

Commit e685ec92 authored by guanyu1's avatar guanyu1
Browse files

rocm_hip修改

parent 472f8d5d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/custom_op.py
View file @ e685ec92
...@@ -184,7 +184,7 @@ class CustomOp(nn.Module): ...@@ -184,7 +184,7 @@ class CustomOp(nn.Module):
return self.maybe_compile(self.forward_native, enable=compile_native) return self.maybe_compile(self.forward_native, enable=compile_native)
if current_platform.is_rocm(): if current_platform.is_rocm():
return self.forward_cuda return self.forward_hip
elif current_platform.is_cpu(): elif current_platform.is_cpu():
return self.forward_cpu return self.forward_cpu
elif current_platform.is_tpu(): elif current_platform.is_tpu():
......
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