iris.x: Device-side communication + .x APIs.

.github/scripts/run_tests.sh

@@ -4,7 +4,7 @@
 #
 # Run Iris tests in a container
 # Usage: run_tests.sh <test_dir> <num_ranks> [gpu_devices] [install_method]
-#   test_dir: subdirectory under tests/ (e.g., examples, unittests, ccl)
+#   test_dir: subdirectory under tests/ (e.g., examples, unittests, ccl, x)
 #   num_ranks: number of GPU ranks (1, 2, 4, or 8)
 #   gpu_devices: comma-separated GPU device IDs (optional)
 #   install_method: pip install method - "git", "editable", or "install" (optional, default: "editable")
@@ -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, or ccl"
+    echo "  test_dir: examples, unittests, ccl, or x"
     echo "  num_ranks: 1, 2, 4, or 8"
     echo "  install_method: git, editable, or install (default: editable)"
     exit 1
@@ -68,6 +68,23 @@ fi
     echo \"Installing iris using method: $INSTALL_METHOD\"
     $INSTALL_CMD
 
+    # Install tritonBLAS if testing x directory (required for GEMM tests)
+    if [ \"$TEST_DIR\" = \"x\" ]; then
+        echo \"Installing tritonBLAS for iris.x GEMM tests...\"
+        # Try to clone and install in editable mode (more reliable)
+        if [ ! -d \"/tmp/tritonBLAS\" ]; then
+            echo \"Cloning tritonBLAS repository...\"
+            cd /tmp && git clone https://github.com/ROCm/tritonBLAS.git 2>&1 | tail -3
+        fi
+        if [ -d \"/tmp/tritonBLAS\" ]; then
+            echo \"Installing tritonBLAS in editable mode...\"
+            pip install -e /tmp/tritonBLAS 2>&1 | tail -3
+        else
+            echo \"Warning: Could not clone tritonBLAS, trying pip install from git...\"
+            pip install git+https://github.com/ROCm/tritonBLAS.git 2>&1 | tail -3 || echo \"Warning: Could not install tritonBLAS\"
+        fi
+    fi
+
     # Run tests in the specified directory
     for test_file in tests/$TEST_DIR/test_*.py; do
         if [ -f \"\$test_file\" ]; then

.github/workflows/iris-tests.yml

@@ -82,6 +82,15 @@ jobs:
           - test_dir: ccl
             num_ranks: 8
             gpu_devices: "0,1,2,3,4,5,6,7"
+          - test_dir: x
+            num_ranks: 1
+            gpu_devices: "0,1"
+          - test_dir: x
+            num_ranks: 2
+            gpu_devices: "2,3"
+          - test_dir: x
+            num_ranks: 4
+            gpu_devices: "4,5,6,7"
 
     steps:
       - name: Checkout repository
@@ -151,6 +160,15 @@ jobs:
           - test_dir: ccl
             num_ranks: 8
             gpu_devices: "0,1,2,3,4,5,6,7"
+          - test_dir: x
+            num_ranks: 1
+            gpu_devices: "0,1"
+          - test_dir: x
+            num_ranks: 2
+            gpu_devices: "2,3"
+          - test_dir: x
+            num_ranks: 4
+            gpu_devices: "4,5,6,7"
 
     steps:
       - name: Checkout repository
@@ -217,6 +235,15 @@ jobs:
           - test_dir: ccl
             num_ranks: 8
             gpu_devices: "0,1,2,3,4,5,6,7"
+          - test_dir: x
+            num_ranks: 1
+            gpu_devices: "0,1"
+          - test_dir: x
+            num_ranks: 2
+            gpu_devices: "2,3"
+          - test_dir: x
+            num_ranks: 4
+            gpu_devices: "4,5,6,7"
 
     steps:
       - name: Checkout repository

.gitignore

@@ -28,4 +28,19 @@ slurm*.out
 
 examples/gemm/results/*
 asm/
-*.img
+*.img
+
+.cache/
+.local/
+.triton/
+.pytest_cache/
+.ruff_cache/
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+*.pyw
+*.pyz
+*.pywz
+*.pyzw
+*.pyzwz