Add files via upload

testzluda.py

@@ -0,0 +1,19 @@
+import torch
+
+try:
+    # Check if CUDA is available
+    if not torch.cuda.is_available():
+        raise RuntimeError("CUDA is not available")
+    
+    # Create tensors on the GPU
+    ten1 = torch.randn((2, 4), device="cuda")
+    ten2 = torch.randn((4, 8), device="cuda")
+    
+    # Perform matrix multiplication
+    output = torch.mm(ten1, ten2)
+    
+    # Print success message
+    print("zluda is correctly installed and working")
+except Exception as e:
+    # Print failure message if any exception occurs
+    print(f"zluda isn't installed correctly: {e}")