From a3aa2d201ecfdc04ff488b7c884d6859118ae214 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sat, 16 May 2026 02:55:25 +0000
Subject: [PATCH] fix: clarify import path setup for CuTeDSL

---
 tests/test_cutedsl.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/test_cutedsl.py b/tests/test_cutedsl.py
index c693acfe..b7df6180 100644
--- a/tests/test_cutedsl.py
+++ b/tests/test_cutedsl.py
@@ -14,11 +14,15 @@ import os
 import math
 import torch
 
-# Add CuTeDSL examples to path
+# Add CuTeDSL examples to path — must be run from the examples directory
+# so that 'from blackwell.kernel.moe.xxx' imports work
 CUTLASS_ROOT = os.environ.get("CUTLASS_ROOT", "/root/cutlass")
 CUTEDSL_EXAMPLES = os.path.join(CUTLASS_ROOT, "examples/python/CuTeDSL")
 sys.path.insert(0, CUTEDSL_EXAMPLES)
 
+# The imports below use 'from blackwell.kernel...' which resolves
+# relative to CUTEDSL_EXAMPLES/cute/ (where the blackwell/ package lives)
+
 import cutlass
 import cutlass.cute as cute
 import cutlass.torch as cutlass_torch