From ad45217102f717309f748d65fc469fff3ef197ad Mon Sep 17 00:00:00 2001
From: chalsliu <45041955+chalsliu@users.noreply.github.com>
Date: Wed, 29 Mar 2023 10:13:11 +0800
Subject: [PATCH] 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
---
 cmake/phi.cmake | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/cmake/phi.cmake b/cmake/phi.cmake
index b2eaf554d2d..be477a30f22 100644
--- a/cmake/phi.cmake
+++ b/cmake/phi.cmake
@@ -82,8 +82,15 @@ function(kernel_declare TARGET_LIST)
         first_registry
         "${kernel_impl}")
     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
-      if(WITH_ROCM OR WITH_NV_JETSON)
+      if(WITH_ROCM)
         string(FIND "${first_registry}" "cuda_only" pos)
         if(pos GREATER 1)
           continue()
-- 
GitLab