Merge branch 'hp/tiles-4' into hp/tiles-5

# Conflicts:
#	docs/source/user_guide/tile16.md
#	tests/python/test_tile16.py

Author: hughperkins

python/quadrants/lang/ast/ast_transformer.py

@@ -40,6 +40,9 @@
 from quadrants.lang.matrix import Matrix, MatrixType
 from quadrants.lang.snode import append, deactivate, length
 from quadrants.lang.struct import Struct, StructType
+from quadrants.lang.util import (
+    is_from_quadrants_module as _is_from_quadrants_module,
+)
 from quadrants.lang.util import (
     is_quadrants_internal_file as _is_quadrants_internal_file,
 )
@@ -678,7 +681,7 @@ def build_Attribute(ctx: ASTTransformerFuncContext, node: ast.Attribute):
                         violation = False
                     if violation and node.value.ptr in [qd_math, math, np]:
                         violation = False
-                    if violation and _is_quadrants_internal_file(ctx.file):
+                    if violation and _is_from_quadrants_module(node.value.ptr):
                         violation = False
                     if violation:
                         message = f"[PURE.VIOLATION] WARNING: Accessing global var {node.attr} from outside function scope within pure kernel {node.value.violates_pure_reason}"

python/quadrants/lang/util.py

@@ -366,4 +366,14 @@ def is_quadrants_internal_file(filepath: str) -> bool:
     return os.path.realpath(filepath).startswith(_quadrants_package_dir() + os.sep)
 
 
+def is_from_quadrants_module(obj: object) -> bool:
+    """Return True if obj belongs to the quadrants package (module, class, or instance)."""
+    import types  # pylint: disable=C0415
+
+    if isinstance(obj, types.ModuleType):
+        return getattr(obj, "__name__", "").startswith("quadrants")
+    mod = getattr(obj, "__module__", None) or getattr(type(obj), "__module__", None)
+    return mod is not None and mod.startswith("quadrants")
+
+
 __all__ = []

tests/python/quadrants/lang/fast_caching/test_pure_validation.py

@@ -251,3 +251,34 @@ def k2() -> qd.f32:
 
     with pytest.warns(UserWarning, match=r"\[PURE\.VIOLATION\]"):
         assert k2() == 23
+
+
+@test_utils.test()
+def test_pure_validation_quadrants_module_attribute_allowed():
+    """Accessing an int constant from the quadrants package should not trigger a purity violation."""
+
+    @qd.kernel(pure=True)
+    def k1() -> qd.i32:
+        return qd.simt.Tile16x16.SIZE
+
+    assert k1() == 16
+
+
+@test_utils.test()
+def test_pure_validation_non_quadrants_attribute_warns():
+    """Accessing an int attribute on a non-quadrants object should still warn/error."""
+    assert qd.lang is not None
+    arch = qd.lang.impl.current_cfg().arch
+    qd.init(arch=arch, offline_cache=False)
+
+    class Config:
+        SIZE = 32
+
+    cfg = Config()
+
+    @qd.kernel(pure=True)
+    def k1() -> qd.i32:
+        return cfg.SIZE
+
+    with pytest.warns(UserWarning, match=r"\[PURE\.VIOLATION\]"):
+        assert k1() == 32