wehub-resource-sync
ba4be087d5
Create PR to main with cherry-pick from release / cherry-pick (push) Failing after 0s
CICD NeMo / pre-flight (push) Failing after 0s
CICD NeMo / configure (push) Has been skipped
Build, validate, and release Neural Modules / pre-flight (push) Failing after 1s
CICD NeMo / code-linting (push) Has been skipped
Build, validate, and release Neural Modules / release (push) Has been skipped
Build, validate, and release Neural Modules / release-summary (push) Has been cancelled
CICD NeMo / cicd-test-container-build (push) Has been cancelled
CICD NeMo / cicd-import-tests (push) Has been cancelled
CICD NeMo / L0_Setup_Test_Data_And_Models (push) Has been cancelled
CICD NeMo / cicd-main-unit-tests (push) Has been cancelled
CICD NeMo / cicd-main-speech (push) Has been cancelled
CICD NeMo / Nemo_CICD_Test (push) Has been cancelled
CICD NeMo / Coverage (e2e) (push) Has been cancelled
CICD NeMo / Coverage (unit-test) (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
CICD NeMo / cicd-wait-in-queue (push) Has been cancelled
25 lines
940 B
Diff
25 lines
940 B
Diff
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
|
|
index d82bebb73c..b52668a54b 100644
|
|
--- a/cpp/CMakeLists.txt
|
|
+++ b/cpp/CMakeLists.txt
|
|
@@ -377,6 +377,7 @@ print(os.path.dirname(torch.__file__),end='');"
|
|
set(nvtx3_dir ${3RDPARTY_DIR}/NVTX/include)
|
|
find_package(Torch REQUIRED)
|
|
|
|
+ set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=1")
|
|
message(STATUS "TORCH_CXX_FLAGS: ${TORCH_CXX_FLAGS}")
|
|
add_compile_options(${TORCH_CXX_FLAGS})
|
|
add_compile_definitions(TORCH_CUDA=1)
|
|
diff --git a/docker/common/install_tensorrt.sh b/docker/common/install_tensorrt.sh
|
|
index 8dda255229..98bdae58f3 100644
|
|
--- a/docker/common/install_tensorrt.sh
|
|
+++ b/docker/common/install_tensorrt.sh
|
|
@@ -28,6 +28,7 @@ for i in "$@"; do
|
|
--CUDNN_VER=?*) CUDNN_VER="${i#*=}";;
|
|
--NCCL_VER=?*) NCCL_VER="${i#*=}";;
|
|
--CUBLAS_VER=?*) CUBLAS_VER="${i#*=}";;
|
|
+ --NVRTC_VER=?*) NVRTC_VER="${i#*=}";;
|
|
*) ;;
|
|
esac
|
|
shift
|