matrice-compute 0.1.29__tar.gz → 0.1.30__tar.gz

{matrice_compute-0.1.29 → matrice_compute-0.1.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_compute
-Version: 0.1.29
+Version: 0.1.30
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_compute-0.1.29 → matrice_compute-0.1.30}/matrice_compute.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_compute
-Version: 0.1.29
+Version: 0.1.30
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_compute-0.1.29 → matrice_compute-0.1.30}/src/matrice_compute/scaling.py

@@ -1266,6 +1266,38 @@ class Scaling:
                 if "jetson" in model or "tegra" in model:
                     is_jetson = True
                     gpu_provider = "NVIDIA"
+                    
+                    # Detect specific Jetson model for GPU architecture
+                    if "orin" in model:
+                        if "agx" in model:
+                            gpu_arch_family = "Jetson Orin AGX"
+                        elif "nx" in model:
+                            gpu_arch_family = "Jetson Orin NX"
+                        elif "nano" in model:
+                            gpu_arch_family = "Jetson Orin Nano"
+                        else:
+                            gpu_arch_family = "Jetson Orin"
+                    elif "thor" in model:
+                        gpu_arch_family = "Jetson Thor"
+                    elif "xavier" in model:
+                        if "agx" in model:
+                            gpu_arch_family = "Jetson Xavier AGX"
+                        elif "nx" in model:
+                            gpu_arch_family = "Jetson Xavier NX"
+                        else:
+                            gpu_arch_family = "Jetson Xavier"
+                    elif "nano" in model and "orin" not in model:
+                        gpu_arch_family = "Jetson Nano"
+                    elif "tx2" in model:
+                        gpu_arch_family = "Jetson TX2"
+                    elif "tx1" in model:
+                        gpu_arch_family = "Jetson TX1"
+                    else:
+                        gpu_arch_family = "Jetson (Unknown Model)"
+                    
+                    # Set gpu_arch to the full model string for detailed info
+                    gpu_arch = model.strip()
+                    
                     try:
                         cuda_result = subprocess.run(["nvcc", "--version"], capture_output=True, text=True)
                         if cuda_result.returncode == 0:
@@ -1275,6 +1307,15 @@ class Scaling:
                                     break
                     except Exception:
                         pass
+                    # Fallback to nvidia-smi for CUDA version if nvcc failed
+                    if not cuda_version:
+                        try:
+                            nvidia_smi_result = subprocess.run(["nvidia-smi", "--query-gpu=driver_version", "--format=csv,noheader"], capture_output=True, text=True)
+                            if nvidia_smi_result.returncode == 0:
+                                # nvidia-smi doesn't directly give CUDA version, but we can infer it's available
+                                cuda_version = "Available (via nvidia-smi)"
+                        except Exception:
+                            pass
         except Exception:
             pass