From 5e77346fc603ec7e37247f3cd68a82636708baa9 Mon Sep 17 00:00:00 2001
From: Weilong Wu <veyron_wu@163.com>
Date: Tue, 6 Sep 2022 13:10:26 +0800
Subject: [PATCH] [Phi] fix phi compile option for jetson (#45757)

* [Phi] fix phi compile option for jetson

* Update phi.cmake
---
 cmake/phi.cmake | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmake/phi.cmake b/cmake/phi.cmake
index f156859200..d50b2ea101 100644
--- a/cmake/phi.cmake
+++ b/cmake/phi.cmake
@@ -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()
-- 
GitLab