未验证 提交 5e77346f 编写于 作者: W Weilong Wu 提交者: GitHub

[Phi] fix phi compile option for jetson (#45757)

* [Phi] fix phi compile option for jetson

* Update phi.cmake
上级 f6d9ec27
......@@ -83,7 +83,7 @@ function(kernel_declare TARGET_LIST)
"${kernel_impl}")
if(NOT first_registry STREQUAL "")
# some gpu kernel only can run on cuda, not support rocm, so we add this branch
if(WITH_ROCM)
if(WITH_ROCM OR WITH_NV_JETSON)
string(FIND "${first_registry}" "cuda_only" pos)
if(pos GREATER 1)
continue()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册