From 1aa2262a2c5849461dc037ada4590e2be43a6e74 Mon Sep 17 00:00:00 2001
From: project6-dev <dev@project6.ai>
Date: Thu, 13 Aug 2026 04:18:15 +0000
Subject: [PATCH] =?UTF-8?q?diag:=20test=5Ftriton.py=20=E2=80=94=20check=20?=
 =?UTF-8?q?if=20Triton=20works=20on=20BI-V100?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test_triton.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 test_triton.py

diff --git a/test_triton.py b/test_triton.py
new file mode 100644
index 00000000..cb6e4971
--- /dev/null
+++ b/test_triton.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+"""Test Triton availability on BI-V100."""
+import torch
+print(f"CUDA available: {torch.cuda.is_available()}")
+print(f"Device: {torch.cuda.get_device_name(0)}")
+
+try:
+    import triton
+    import triton.language as tl
+    print(f"Triton version: {triton.__version__}")
+
+    @triton.jit
+    def add_kernel(x_ptr, y_ptr, out_ptr, n, BLOCK: tl.constexpr):
+        pid = tl.program_id(0)
+        offs = pid * BLOCK + tl.arange(0, BLOCK)
+        mask = offs < n
+        x = tl.load(x_ptr + offs, mask=mask)
+        y = tl.load(y_ptr + offs, mask=mask)
+        tl.store(out_ptr + offs, x + y, mask=mask)
+
+    n = 1024
+    x = torch.randn(n, device="cuda")
+    y = torch.randn(n, device="cuda")
+    out = torch.empty(n, device="cuda")
+    grid = lambda meta: (triton.cdiv(n, meta['BLOCK']),)
+    add_kernel[grid](x, y, out, n, BLOCK=256)
+    torch.cuda.synchronize()
+    ref = x + y
+    diff = (out - ref).abs().max().item()
+    print(f"Triton kernel test: diff={diff:.8f} {'PASS' if diff < 1e-6 else 'FAIL'}")
+except ImportError as e:
+    print(f"Triton not available: {e}")
+except Exception as e:
+    print(f"Triton error: {e}")