chore: Enable debug prints and fix diagnostic script

Author: drtobybrown

benchmarks/verify_gpu_fallback.py

@@ -60,6 +60,8 @@ def verify_gpu_fallback():
     # DIAGNOSTIC: Test TorchWrapper directly
     print("\n--- TorchWrapper Diagnostic ---")
     try:
+        import torch
+        import numpy as np
         try:
             from ppxf.torch_wrapper import TorchWrapper
         except ImportError:
@@ -74,6 +76,9 @@ def verify_gpu_fallback():
         z_float = tw.zeros((10, 10), dtype=float)
         print(f"tw.zeros((10,10), dtype=float) dtype: {z_float.dtype}")
 
+        z_np = tw.zeros((10, 10), dtype=np.float64)
+        print(f"tw.zeros((10,10), dtype=np.float64) dtype: {z_np.dtype}")
+        
         if z.dtype == torch.float32 and tw.device.type != 'mps':
             print("CRITICAL IF: TorchWrapper defaulting to float32 on non-MPS device!")
 

torch_wrapper.py

@@ -64,6 +64,8 @@ def asnumpy(self, a):
     def zeros(self, shape, dtype=float):
         # Default to float (float64) to match dict(numpy)
         t_dtype = self._get_dtype(dtype)
+        # DEBUG
+        print(f"DEBUG: TorchWrapper.zeros requested dtype={dtype} mapped to {t_dtype} on device {self.device}")
         return torch.zeros(shape, device=self.device, dtype=t_dtype)
 
     def ones(self, shape, dtype=float):
@@ -79,11 +81,20 @@ def _map_dtype(self, dtype):
         if dtype == int: return torch.long
         if dtype == float: return torch.float64 
         if dtype == complex: return torch.complex128
+        # Handle pure numpy types if passed
+        if dtype == np.float64: return torch.float64
+        if dtype == np.float32: return torch.float32
+        if dtype == np.int64: return torch.long
+        if dtype == np.int32: return torch.int32
+        if dtype == np.complex128: return torch.complex128
+        if dtype == np.complex64: return torch.complex64
         return dtype
 
     def _get_dtype(self, dtype):
         t_dtype = self._map_dtype(dtype)
-        return self._apply_device_constraints(t_dtype)
+        res = self._apply_device_constraints(t_dtype)
+        # print(f"DEBUG: _get_dtype({dtype}) -> mapped {t_dtype} -> constrained {res}")
+        return res
 
     def _apply_device_constraints(self, dtype):
         if self.device.type == 'mps':