Skip to content

Commit c834302

Browse files
committed
Merge with dev
1 parent 2c28b72 commit c834302

2 files changed

Lines changed: 14 additions & 1 deletion

File tree

3rdparty/cudnn-frontend

Submodule cudnn-frontend updated 159 files

buildit.sh

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
#!/bin/bash
2+
3+
export NVTE_FRAMEWORK=pytorch
4+
export NVTE_ROCM_ARCH=gfx950
5+
export NVTE_USE_ROCM=1
6+
export CU_NUM=256
7+
8+
pip install -U ninja psutil pybind11
9+
10+
export NVTE_AITER_PREBUILT_BASE_URL=https://compute-artifactory.amd.com:5000/artifactory/rocm-generic-local/te-ci/aiter-prebuilts
11+
12+
pip install -ve . --no-build-isolation
13+

0 commit comments

Comments
 (0)