Update torch_vgpu.py

torch_vgpu.py

@@ -18,21 +18,21 @@ def init_vgpu_backend():
             # First rename privateuse1 to vgpu
             torch.utils.rename_privateuse1_backend("vgpu")
             
-            # Create library for the backend
-            lib = Library("vgpu", "IMPL")
+            # Create an aten library implementation for our backend
+            lib = Library("aten", "IMPL", "vgpu")
             
             # Register essential operations for the backend
-            @impl(lib, "aten::empty.memory_format")
+            @impl(lib, "empty.memory_format")
             def empty_impl(size, dtype=None, layout=None, device=None, pin_memory=None, memory_format=None):
                 # Create empty tensor on CPU first, will be moved to vGPU storage later
                 return torch.empty(size, dtype=dtype, device='cpu')
             
-            @impl(lib, "aten::copy.from")
+            @impl(lib, "copy.from")
             def copy_impl(self, src, non_blocking=False):
                 # Handle tensor copying between devices
                 return torch.tensor(src.cpu().numpy(), device='cpu')
             
-            @impl(lib, "aten::_to_copy")
+            @impl(lib, "_to_copy")
             def to_impl(self, dtype=None, device=None, non_blocking=False, copy=False):
                 # Handle tensor device transfer
                 if device and str(device).startswith("vgpu"):