From e5e9b891e14c922b48dac4c2d6fd59e85d554e8b Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Tue, 13 Jan 2026 13:53:48 -0500
Subject: [PATCH] [TEST] Introduce release extra tests

This PR introduces release extra tests to help run some integration
of downstream packages, making it easy to run quick checks during release.
---
 tests/release-extra/README.md                | 30 +++++++++++
 tests/release-extra/test_cute_dsl_compile.py | 56 ++++++++++++++++++++
 2 files changed, 86 insertions(+)
 create mode 100644 tests/release-extra/README.md
 create mode 100644 tests/release-extra/test_cute_dsl_compile.py

diff --git a/tests/release-extra/README.md b/tests/release-extra/README.md
new file mode 100644
index 00000000..5e397d2b
--- /dev/null
+++ b/tests/release-extra/README.md
@@ -0,0 +1,30 @@
+<!--- Licensed to the Apache Software Foundation (ASF) under one -->
+<!--- or more contributor license agreements.  See the NOTICE file -->
+<!--- distributed with this work for additional information -->
+<!--- regarding copyright ownership.  The ASF licenses this file -->
+<!--- to you under the Apache License, Version 2.0 (the -->
+<!--- "License"); you may not use this file except in compliance -->
+<!--- with the License.  You may obtain a copy of the License at -->
+
+<!---   http://www.apache.org/licenses/LICENSE-2.0 -->
+
+<!--- Unless required by applicable law or agreed to in writing, -->
+<!--- software distributed under the License is distributed on an -->
+<!--- "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY -->
+<!--- KIND, either express or implied.  See the License for the -->
+<!--- specific language governing permissions and limitations -->
+<!--- under the License. -->
+
+# Extra integration tests for release
+
+This folder contains an extra set of regression tests related to important
+downstream integration use cases that can be run during release to test integration compatibility of the code.
+While most of the tests should be covered by downstream tests and unit tests (UT), it is still helpful to have some
+form of quick integration checks during release.
+
+**High-level summary:**
+
+- This folder should not be used in CI (as it integrates downstream usage tests).
+- We aim to curate this folder to contain a reasonably minimal set
+  that covers representative use cases based on regressions and lessons learned.
+- It is not meant to replace downstream tests, but can help catch issues during release.
diff --git a/tests/release-extra/test_cute_dsl_compile.py b/tests/release-extra/test_cute_dsl_compile.py
new file mode 100644
index 00000000..5f75c10a
--- /dev/null
+++ b/tests/release-extra/test_cute_dsl_compile.py
@@ -0,0 +1,56 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from __future__ import annotations
+
+import cutlass
+import pytest
+import torch
+from cutlass import cute
+
+
+@cute.kernel
+def device_add_one(a: cute.Tensor) -> None:
+    a[0] += cutlass.Float16(1.0)
+
+
+@cute.jit
+def add_one(a: cute.Tensor) -> None:
+    device_add_one(a).launch(grid=(1, 1, 1), block=(1, 1, 1))
+
+
+def test_cute_dsl_compile() -> None:
+    dtype = cutlass.Float16
+    alignment_bytes = 16
+    divisibility = alignment_bytes * 8 // dtype.width
+
+    fake_tensor = cute.runtime.make_fake_compact_tensor(
+        dtype,
+        (cute.SymInt(), cute.SymInt(divisibility=divisibility)),
+        stride_order=(1, 0),
+        assumed_align=alignment_bytes,
+    )
+
+    compiled_add_one = cute.compile(add_one, fake_tensor, options="--enable-tvm-ffi")
+    # Pass in tensor with correct divisibility
+    a = torch.zeros((4, 16), device="cuda", dtype=torch.float16)
+    compiled_add_one(a)
+
+    # Pass in tensor with incorrect divisibility
+    with pytest.raises(ValueError):
+        b = torch.zeros((4, 18), device="cuda", dtype=torch.float16)
+        compiled_add_one(b)