use cpu demos

ivy-llc · Jun 4, 2024 · 5bbb80a · 5bbb80a
1 parent 998084b
commit 5bbb80a
Show file tree

Hide file tree

Showing 11 changed files with 6,778 additions and 7 deletions.
diff --git a/.github/workflows/test-nightly-binaries.yml b/.github/workflows/test-nightly-binaries.yml
@@ -10,16 +10,16 @@ jobs:
       fail-fast: false
       matrix:
         module: [
-          alexnet_demo,
+          alexnet_demo_cpu,
           # bert_demo, # tests disabled until the OOM issue is resolved
-          image_segmentation_with_ivy_unet,
+          image_segmentation_with_ivy_unet_cpu,
           # mmpretrain_to_jax, # tests disabled until we get cuda installation working in the docker image
-          resnet_demo,
-          torch_to_jax,
-          lstm_tensorflow_to_torch,
-          lstm_torch_to_tensorflow,
+          resnet_demo_cpu,
+          torch_to_jax_cpu,
+          lstm_tensorflow_to_torch_cpu,
+          lstm_torch_to_tensorflow_cpu,
           # dinov2_to_paddle, # tests disabled until we get cuda installation working in the docker image
-          convnext_to_torch,
+          convnext_to_torch_cpu,
           # xgboost_demo, # tests disabled until we get the csv files uploaded to drive for being used in the tests
         ]
 

diff --git a/examples_and_demos/alexnet_demo_cpu.ipynb b/examples_and_demos/alexnet_demo_cpu.ipynb