23 lines
501 B
Bash
23 lines
501 B
Bash
#!/bin/bash
|
|
|
|
set -eo pipefail
|
|
set -x
|
|
: ${MAX_JOBS:=$(nproc)}
|
|
: ${CUDA_VISIBLE_DEVICES:=0}
|
|
|
|
pip install -e . -v
|
|
pip install --upgrade nvidia-cudnn-cu12
|
|
pip install --upgrade cuda-python==12.*
|
|
|
|
# run task_blackwell_utils_kernels.sh
|
|
bash scripts/run_test_blackwell_utils_kernels.sh
|
|
|
|
# run task_blackwell_attention_kernels.sh
|
|
bash scripts/run_test_blackwell_attention_kernels.sh
|
|
|
|
# gemm kernels
|
|
bash scripts/run_test_blackwell_gemm_kernels.sh
|
|
|
|
# moe kernels
|
|
bash scripts/run_test_blackwell_moe_kernels.sh
|