Install numba

triton-lang · Oct 17, 2023 · c66c198 · c66c198
1 parent fb55916
commit c66c198
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml
@@ -35,6 +35,7 @@ jobs:
         run: |
           
           python3 -m pip install --upgrade pip
+          python3 -m pip install numba
           python3 -m pip install --upgrade torch==2.2.0.dev20231017 torchvision==0.17.0.dev20231017 torchaudio==2.2.0.dev20231017 --index-url https://download.pytorch.org/whl/nightly/cu121
           cd torchbench
           python3 install.py -v BERT_pytorch