Fix CI not installing rocThrust

Fix CI CUDA flag version
This commit is contained in:
Tom Lin 2021-11-12 00:14:07 +00:00
parent a463e88895
commit 0d55a7261b
2 changed files with 2 additions and 2 deletions

View File

@ -254,7 +254,7 @@ setup_rocm() {
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
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 HIP_CXX "$ROCM_PATH/bin/hipcc"
verify_bin_exists "$HIP_CXX"

View File

@ -124,7 +124,7 @@ run_build() {
AMD_ARCH="gfx_903"
NV_ARCH="sm_70"
NV_ARCH_CCXY="cuda11.3,cc80"
NV_ARCH_CCXY="cuda11.4,cc80"
build_gcc() {
local name="gcc_build"