Fix CI not installing rocThrust
Fix CI CUDA flag version
This commit is contained in:
parent
a463e88895
commit
0d55a7261b
@ -254,7 +254,7 @@ setup_rocm() {
|
|||||||
wget -q -O - "https://repo.radeon.com/rocm/rocm.gpg.key" | sudo apt-key add -
|
wget -q -O - "https://repo.radeon.com/rocm/rocm.gpg.key" | sudo apt-key add -
|
||||||
echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/4.5 ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list
|
echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/4.5 ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list
|
||||||
sudo apt-get update -qq
|
sudo apt-get update -qq
|
||||||
sudo apt-get install -y -qq rocm-dev
|
sudo apt-get install -y -qq rocm-dev rocthrust-dev
|
||||||
export_var ROCM_PATH "/opt/rocm"
|
export_var ROCM_PATH "/opt/rocm"
|
||||||
export_var HIP_CXX "$ROCM_PATH/bin/hipcc"
|
export_var HIP_CXX "$ROCM_PATH/bin/hipcc"
|
||||||
verify_bin_exists "$HIP_CXX"
|
verify_bin_exists "$HIP_CXX"
|
||||||
|
|||||||
@ -124,7 +124,7 @@ run_build() {
|
|||||||
|
|
||||||
AMD_ARCH="gfx_903"
|
AMD_ARCH="gfx_903"
|
||||||
NV_ARCH="sm_70"
|
NV_ARCH="sm_70"
|
||||||
NV_ARCH_CCXY="cuda11.3,cc80"
|
NV_ARCH_CCXY="cuda11.4,cc80"
|
||||||
|
|
||||||
build_gcc() {
|
build_gcc() {
|
||||||
local name="gcc_build"
|
local name="gcc_build"
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user