rocm job failed at merge time, using export LD_LIBRARY_PATH

divyanshk · divyanshk · commit b0b06d3bff6a · 2026-03-11T10:51:45.000-07:00
diff --git a/.ci/pytorch/test.sh b/.ci/pytorch/test.sh
@@ -1156,16 +1156,14 @@ test_libtorch_jit() {
 
 test_libtorch_profiler() {
   echo "Testing profiler C++ tests"
-  ln -sf "$TORCH_LIB_DIR"/libc10* "$TORCH_BIN_DIR"
-  ln -sf "$TORCH_LIB_DIR"/libtorch* "$TORCH_BIN_DIR"
+  export CPP_TESTS_DIR="${TORCH_BIN_DIR}"
+  export LD_LIBRARY_PATH="${TORCH_LIB_DIR}:${LD_LIBRARY_PATH}"
 
-  # Run E2E test in its own process (needs clean Kineto state)
-  "${TORCH_BIN_DIR}/test_privateuse1_profiler" \
-    --gtest_filter=PrivateUse1ProfilerTest.EndToEndProfiling
+  # Run E2E test first (needs clean Kineto state)
+  python test/run_test.py --cpp --verbose -i cpp/test_privateuse1_profiler -k "EndToEndProfiling"
 
-  # Run all other tests in a separate process
-  "${TORCH_BIN_DIR}/test_privateuse1_profiler" \
-    --gtest_filter=-PrivateUse1ProfilerTest.EndToEndProfiling
+  # Run all other tests
+  python test/run_test.py --cpp --verbose -i cpp/test_privateuse1_profiler -k "not EndToEndProfiling"
 }
 
 test_libtorch_api() {
