test cuda

2024-08-02 14:58:19 +01:00
parent 7599df7ffd
commit 3313151010
1 changed files with 9 additions and 0 deletions
@@ -171,6 +171,14 @@ jobs:
          inference_engine: mlx
          model_id: llama-3.1-8b

+  test_cuda:
+    machine:
+      image: linux-cuda-12:default
+    resource_class: gpu.nvidia.medium
+    steps:
+      - checkout
+      - run: nvidia-smi
+
  chatgpt_api_integration_test_tinygrad:
    machine:
      image: linux-cuda-12:default
@@ -204,4 +212,5 @@ workflows:
      - unit_test
      - discovery_integration_test
      - chatgpt_api_integration_test_mlx
+      - test_cuda
      - chatgpt_api_integration_test_tinygrad