PaddlePaddle
diff --git a/‎paddle/phi/CMakeLists.txt
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions b/‎paddle/phi/CMakeLists.txt
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions
diff --git a/‎test/legacy_test/CMakeLists.txt
Copy file name to clipboard
+7Lines changed: 7 additions & 0 deletions b/‎test/legacy_test/CMakeLists.txt
Copy file name to clipboard
+7Lines changed: 7 additions & 0 deletions
@@ -198,6 +198,8 @@ add_library(phi ${PHI_BUILD_TYPE} ${PHI_DUMMY_FILE})
 target_link_libraries(phi phi_core)
 if(WITH_GPU OR WITH_ROCM)
   target_link_libraries(phi phi_gpu)
+  target_link_libraries(phi_gpu ${ROCM_HIPRTC_LIB})
+  target_link_libraries(phi_core ${ROCM_HIPRTC_LIB})
 endif()
 
 # Note(silverling): some functions in phi_core depend on phi_gpu,
 
@@ -99,6 +99,13 @@ if(((NOT WITH_ROCM) AND (NOT WITH_GPU)) OR WIN32)
   list(REMOVE_ITEM TEST_OPS test_async_offload_reload)
 endif()
 
+if(WITH_ROCM)
+  add_test(NAME test_hygon_llama_ops
+           COMMAND bash
+                   ${CMAKE_CURRENT_SOURCE_DIR}/hygon_dcu/test_hygon_llama_op.sh)
+  set_tests_properties(test_hygon_llama_ops PROPERTIES TIMEOUT 600)
+endif()
+
 if(NOT WITH_NCCL)
   list(REMOVE_ITEM TEST_OPS test_async_offload_reload)
 endif()