Absorb iris.x into Context, delete iris/x/ (#521)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

Author: mawad-amd

.github/scripts/run_tests.sh

@@ -22,7 +22,7 @@ INSTALL_METHOD=${4:-"editable"}
 if [ -z "$TEST_DIR" ] || [ -z "$NUM_RANKS" ]; then
     echo "[ERROR] Missing required arguments"
     echo "Usage: $0 <test_dir> <num_ranks> [gpu_devices] [install_method]"
-    echo "  test_dir: examples, unittests, x or ccl"
+    echo "  test_dir: examples, unittests, context, ccl, or ops"
     echo "  num_ranks: 1, 2, 4, or 8"
     echo "  install_method: git, editable, or install (default: editable)"
     exit 1

.github/workflows/iris-nightly-triton-test.yml

@@ -46,13 +46,13 @@ jobs:
             num_ranks: 4
           - test_dir: ccl
             num_ranks: 8
-          - test_dir: x
+          - test_dir: context
             num_ranks: 1
-          - test_dir: x
+          - test_dir: context
             num_ranks: 2
-          - test_dir: x
+          - test_dir: context
             num_ranks: 4
-          - test_dir: x
+          - test_dir: context
             num_ranks: 8
           - test_dir: ops
             num_ranks: 1

.github/workflows/iris-tests.yml

@@ -48,13 +48,13 @@ jobs:
             num_ranks: 4
           - test_dir: ccl
             num_ranks: 8
-          - test_dir: x
+          - test_dir: context
             num_ranks: 1
-          - test_dir: x
+          - test_dir: context
             num_ranks: 2
-          - test_dir: x
+          - test_dir: context
             num_ranks: 4
-          - test_dir: x
+          - test_dir: context
             num_ranks: 8
           - test_dir: ops
             num_ranks: 1
@@ -124,7 +124,7 @@ jobs:
             num_ranks: 2
           - test_dir: ccl
             num_ranks: 2
-          - test_dir: x
+          - test_dir: context
             num_ranks: 2
           - test_dir: ops
             num_ranks: 2
@@ -185,7 +185,7 @@ jobs:
             num_ranks: 2
           - test_dir: ccl
             num_ranks: 2
-          - test_dir: x
+          - test_dir: context
             num_ranks: 2
           - test_dir: ops
             num_ranks: 2

iris/__init__.py

@@ -45,6 +45,13 @@
 from iris.host.iris import Iris, iris
 from iris.mem.triton.context import Context, Context as DeviceContext
 from iris.host.tracing.events import TraceEvent
+from iris.mem.triton.types import (
+    Tile,
+    TileView,
+    TensorView,
+    AllReduceConfig,
+    make_tensor_view,
+)
 from iris.mem.triton.ops import (
     load,
     store,
@@ -94,6 +101,11 @@
     "Context",
     "DeviceContext",
     "TraceEvent",
+    "Tile",
+    "TileView",
+    "TensorView",
+    "AllReduceConfig",
+    "make_tensor_view",
     "load",
     "store",
     "copy",

iris/mem/triton/__init__.py

@@ -6,6 +6,19 @@
 from .context import Context as DeviceContext  # noqa: F401  backward compat
 from .tracing import Tracing  # noqa: F401
 from .tracing import Tracing as DeviceTracing  # noqa: F401  backward compat
+from .types import (  # noqa: F401
+    Tile,
+    TileView,
+    TensorView,
+    AllReduceConfig,
+    make_tensor_view,
+    tile_layout,
+    tile_ptr,
+    offset_ptr,
+    chiplet_transform_chunked,
+    compute_tile_indices,
+    compute_tile_offsets,
+)
 from .ops import (  # noqa: F401
     load,
     store,