Rust-related/luminal
1
0
Fork 1
mirror of https://github.com/jafioti/luminal.git synced 2026-06-01 21:49:47 +09:00
Code Issues Packages Projects Releases Wiki Activity

force cuda 12.8 for modal

Browse Source
This commit is contained in:
Austin Glover
2026-03-25 00:12:55 +00:00
parent 638765b62b
commit 5310335256
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
ci/modal_example.py
View File

@@ -4,6 +4,7 @@ import os
example = os.environ.get("EXAMPLE", "llama") example = os.environ.get("EXAMPLE", "llama")
gpu_type = os.environ.get("GPU_TYPE", "A100-80GB") gpu_type = os.environ.get("GPU_TYPE", "A100-80GB")
CUDARC_CUDA_VERSION = "12080"
app = modal.App(f"luminal-ci-{example}") app = modal.App(f"luminal-ci-{example}")
@@ -19,7 +20,12 @@ cuda_image = (
.run_commands( .run_commands(
"curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y", "curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y",
) )
.env({"PATH": "/root/.cargo/bin:$PATH"}) .env(
{
"PATH": "/root/.cargo/bin:$PATH",
"CUDARC_CUDA_VERSION": CUDARC_CUDA_VERSION,
}
)
.add_local_dir(".", remote_path=WORKDIR) .add_local_dir(".", remote_path=WORKDIR)
) )
@@ -41,6 +47,7 @@ def run_example(example: str):
cwd=f"{WORKDIR}/examples/{example}", cwd=f"{WORKDIR}/examples/{example}",
env={ env={
**os.environ, **os.environ,
"CUDARC_CUDA_VERSION": CUDARC_CUDA_VERSION,
"HF_HOME": "/root/.cache/huggingface", "HF_HOME": "/root/.cache/huggingface",
}, },
check=True, check=True,