Skip to content

Commit b7347ce

Browse files
authored
[https://nvbugs/5433581][fix] Revert deep_gemm installation workaround for SBSA (NVIDIA#6666)
Signed-off-by: Yanchao Lu <[email protected]>
1 parent 98424f3 commit b7347ce

File tree

2 files changed

+0
-17
lines changed

2 files changed

+0
-17
lines changed

docs/source/installation/linux.md

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,6 @@
1616
# Optional step: Only required for NVIDIA Blackwell GPUs and SBSA platform
1717
pip3 install torch==2.7.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
1818

19-
# Optional step: Workaround for deep_gemm installation failure on SBSA platform
20-
# The actual deep_gemm package and version should be obtained from the requirements.txt file.
21-
pip3 install 'deep_gemm @ git+https://github.com/zongfeijing/DeepGEMM.git@a9d538ef4dff0326fe521c6ca0bfde115703b56a' \
22-
--extra-index-url https://download.pytorch.org/whl/cu128
23-
2419
sudo apt-get -y install libopenmpi-dev
2520
```
2621

jenkins/L0_Test.groovy

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2106,18 +2106,6 @@ def launchTestJobs(pipeline, testFilter, dockerNode=null)
21062106
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.7.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128")
21072107
}
21082108

2109-
// Workaround for https://nvbugs/5433581 where deep_gemm installation fails on SBSA platform
2110-
if (cpu_arch == AARCH64_TRIPLE) {
2111-
echo "###### Workaround for https://nvbugs/5433581 Start ######"
2112-
def deepGemmLine = readFile("${LLM_ROOT}/requirements.txt").readLines().find { it.trim().startsWith('deep_gemm') }
2113-
if (deepGemmLine) {
2114-
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install '${deepGemmLine.trim()}' --extra-index-url https://download.pytorch.org/whl/cu128")
2115-
}
2116-
else {
2117-
echo "deep_gemm package not found in requirements.txt"
2118-
}
2119-
}
2120-
21212109
def libEnv = []
21222110
if (env.alternativeTRT) {
21232111
stage("Replace TensorRT") {

0 commit comments

Comments
 (0)