test cuda

33131510102446156450c6fc149fbb7357e13520 · 2024-08-02 14:58:19 +0100 · Alex Cheema

Files touched

M .circleci/config.yml

Diff

commit 33131510102446156450c6fc149fbb7357e13520
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Aug 2 14:58:19 2024 +0100

    test cuda
---
 .circleci/config.yml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index cb58bb37..5358caae 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -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
\ No newline at end of file

← 7599df7f trigger circleci · back to Exo · trigger circleci aa99c8ae →