瀏覽代碼

tweak ci for unit tests

Alex Cheema 7 月之前
父節點
當前提交
b1ec5ae298
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      .circleci/config.yml

+ 2 - 2
.circleci/config.yml

@@ -91,7 +91,7 @@ commands:
 jobs:
 jobs:
   unit_test:
   unit_test:
     macos:
     macos:
-      xcode: "15.4.0"
+      xcode: "16.0.0"
     resource_class: macos.m1.medium.gen1
     resource_class: macos.m1.medium.gen1
     steps:
     steps:
       - checkout
       - checkout
@@ -113,7 +113,7 @@ jobs:
             source env/bin/activate
             source env/bin/activate
             # set TEMPERATURE to 0 for deterministic sampling
             # set TEMPERATURE to 0 for deterministic sampling
             echo "Running inference engine tests..."
             echo "Running inference engine tests..."
-            METAL_XCODE=1 TEMPERATURE=0 python3 -m exo.inference.test_inference_engine
+            LOW_MEMORY=1 DEVICE=gpu METAL_DEVICE_WRAPPER_TYPE=1 METAL_DEBUG_ERROR_MODE=0 METAL_XCODE=1 TEMPERATURE=0 python3 -m exo.inference.test_inference_engine
             echo "Running tokenizer tests..."
             echo "Running tokenizer tests..."
             python3 ./test/test_tokenizers.py
             python3 ./test/test_tokenizers.py