onnx testing

Author: randomizedcoder

desktop/l/flake.nix

@@ -37,7 +37,20 @@
         };
         overlays = [
           (final: prev: {
-            onnxruntime = final.callPackage /home/das/Downloads/nixpkgs/pkgs/by-name/on/onnxruntime/package.nix { };
+            onnxruntime = final.callPackage /home/das/Downloads/nixpkgs/pkgs/by-name/on/onnxruntime/package.nix {
+              rocmSupport = true;
+              rcclSupport = true;
+            };
+            python313Packages = prev.python313Packages.override (old: {
+              overrides = prev.lib.composeManyExtensions [
+                (final: prev: {
+                  onnxruntime = final.callPackage /home/das/Downloads/nixpkgs/pkgs/development/python-modules/onnxruntime/default.nix {
+                    onnxruntime = final.onnxruntime;  # Use the overlay version
+                  };
+                })
+                old.overrides or (final: prev: { })
+              ];
+            });
           })
         ];
       };

desktop/l/home.nix

@@ -58,6 +58,9 @@
     #CHROME_EXECUTABLE = "/etc/profiles/per-user/das/bin/google-chrome-stable";
     CHROME_EXECUTABLE = "${pkgs.google-chrome}/bin/google-chrome-stable";
     GOOGLE_APPLICATION_CREDENTIALS="~/Downloads/dashboard-dev-3da32-83d127a0f9ba.json";
+
+    # Python path for onnxruntime
+    PYTHONPATH = "${pkgs.python313Packages.onnxruntime}/lib/python3.13/site-packages";
   };
 
   home.packages = with pkgs; [
@@ -130,7 +133,7 @@
 
     # Scripting/Utils
     perl
-    python314
+    python313
     gawk
     jq
     git
@@ -451,6 +454,9 @@
 
     # Custom onnxruntime package
     onnxruntime
+
+    # Custom Python onnxruntime module
+    python313Packages.onnxruntime
   ];
 
   # vscode
@@ -506,6 +512,7 @@
     };
   };
 
+
   programs.vim = {
     enable = true;
     plugins = with pkgs.vimPlugins; [ vim-airline ];