Fix CI TestChunkedSGMV (#10737)

2025-09-22 16:09:58 +08:00
parent 70e4b21853
commit 2101d93b4f
2 changed files with 17 additions and 0 deletions
@@ -621,6 +621,12 @@ class CachedKernel:

        return complete_args

+    def _clear_cache(self):
+        """
+        Clear the kernel cache for testing purposes.
+        """
+        self.kernel_cache.clear()
+

 def cached_triton_kernel(key_fn=None):
    """