v4.0 update. (#2371)

2025-06-06 14:39:20 +08:00
parent 2e2af190bd
commit 8bdbfca682
254 changed files with 29751 additions and 1980 deletions
--- a/test/unit/nvrtc/thread/nvrtc_contraction.cu
+++ b/test/unit/nvrtc/thread/nvrtc_contraction.cu
@@ -51,6 +51,9 @@ TEST(SM90_nvrtc_kernel, Contraction) {
    "-std=c++17",
    "-arch=sm_90",
    "-I" CUDA_INCLUDE_DIR,
+#if (__CUDACC_VER_MAJOR__ >= 13)
+    "-I" CUDA_INCLUDE_DIR "/cccl",
+#endif // __CUDACC_VER_MAJOR__ >= 13
  };

  EXPECT_TRUE(test::nvrtc::thread::TestbedKernel::compile(
@@ -60,7 +63,7 @@ TEST(SM90_nvrtc_kernel, Contraction) {
        "cute::Shape<cute::_1, cute::_2, cute::_1>,"
        "true, true,"
        "10, 10, 10, 10>::Kernel",
-    { nvrtc_opts, nvrtc_opts + 5 }
+    { std::begin(nvrtc_opts), std::end(nvrtc_opts) }
  ));
 }
 #endif