Initial project scaffold

tests/test_extension_import.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+import pytest
+import torch
+
+from tools.lab_extension import build_extension
+
+
+@pytest.mark.cuda_required
+@pytest.mark.skeleton
+def test_extension_can_build_or_skip():
+    if not torch.cuda.is_available():
+        pytest.skip("CUDA is not available")
+    ext = build_extension(verbose=False)
+    if ext is None:
+        pytest.skip("extension build/load is unavailable in this environment")
+    assert hasattr(torch.ops, "kernel_lab")
+    assert hasattr(torch.ops.kernel_lab, "vector_add")