Update torch_vgpu.py

torch_vgpu.py

@@ -81,7 +81,7 @@ class VGPUDevice:
     def __init__(self, vram: Optional[VirtualVRAM] = None):
         self.vram = vram or VirtualVRAM()
         self.tensor_cores = None  # Will be initialized when needed
-        self.device_name = "privateuseone"  # Our registered device type
+        self.device_name = "vgpu"  # Our registered device type
         self._register_device()
 
     def _register_device(self):
@@ -103,10 +103,16 @@ class VGPUDevice:
                     self.device = device
                 
                 def __call__(self, size, dtype=None, device=None):
-                    # Create tensor on CPU first
-                    cpu_tensor = torch.empty(size, dtype=dtype, device='cpu')
-                    # Move to vGPU storage
-                    return to_vgpu(cpu_tensor, self.vram)
+                    # Create tensor directly in vGPU memory
+                    tensor_id = f"tensor_empty_{id(size)}"
+                    # Initialize empty array of the right size and dtype
+                    shape = size if isinstance(size, (tuple, list)) else (size,)
+                    data = np.empty(shape, dtype=np.float32 if dtype is None else dtype)
+                    # Store directly in vRAM
+                    self.vram.storage.store_tensor(tensor_id, data)
+                    # Create tensor with our device type
+                    result = torch.as_tensor(data, device=self.device)
+                    return result
             
             # Set up allocator
             self._allocator = VGPUAllocator(self.vram, self._device)
@@ -187,4 +193,3 @@ def to_vgpu(tensor: torch.Tensor, vram: Optional[VirtualVRAM] = None) -> torch.T
     # Set the device using the internal name
     result.data = result.data.to(device._device)
     return result
-