fix(tests): properly import conftest helpers

Use try/except pattern to import conftest functions
to support both pytest auto-import and direct module
imports.

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>

Author: shijiashuai

tests/test_attention.py

@@ -12,8 +12,12 @@
 except ImportError:  # pragma: no cover - environment-dependent guard
     pytest.skip("PyTorch is not installed", allow_module_level=True)
 
-# Import from conftest (pytest automatically makes conftest functions available)
-from conftest import assert_close, compute_attention_reference
+# Import helper functions from conftest
+try:
+    from conftest import assert_close, compute_attention_reference
+except ImportError:
+    # Fallback for direct module imports
+    from tests.conftest import assert_close, compute_attention_reference
 
 # Strategies for property-based testing
 batch_strategy = st.integers(min_value=1, max_value=4)

tests/test_gemm.py

@@ -12,8 +12,12 @@
 except ImportError:  # pragma: no cover - environment-dependent guard
     pytest.skip("PyTorch is not installed", allow_module_level=True)
 
-# Import from conftest (pytest automatically makes conftest functions available)
-from conftest import assert_close
+# Import helper functions from conftest
+try:
+    from conftest import assert_close
+except ImportError:
+    # Fallback for direct module imports
+    from tests.conftest import assert_close
 
 # Strategies for property-based testing
 dim_strategy = st.integers(min_value=16, max_value=512)

tests/test_interface.py

@@ -12,8 +12,12 @@
 except ImportError:  # pragma: no cover - environment-dependent guard
     pytest.skip("PyTorch is not installed", allow_module_level=True)
 
-# Import from conftest (pytest automatically makes conftest functions available)
-from conftest import assert_close
+# Import helper functions from conftest
+try:
+    from conftest import assert_close
+except ImportError:
+    # Fallback for direct module imports
+    from tests.conftest import assert_close
 
 
 class TestPythonInterface:

tests/test_profiler.py

@@ -10,7 +10,7 @@
 except ImportError:
     pytest.skip("CUDA kernels not built", allow_module_level=True)
 
-from python.profiler import (
+from cuda_llm_ops.profiler import (
     Bottleneck,
     CUDAProfiler,
     KernelMetrics,