未验证 提交 ad452171 编写于 作者: C chalsliu 提交者: GitHub

Fix jetson conv2d_fusion not found error (#51654)

* Fix jetson conv2d_fusion not found error

* Fix jetson conv2d_fusion not found error

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