Exclude blkq4_fp16_gemm_sm80_test in cuda 12.5 build (#21373)

There is build errors when build with CUDA 12.5 and
`--cmake_extra_defines onnxruntime_BUILD_UNIT_TESTS=ON
onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON`.

Temporally exclude blkq4_fp16_gemm_sm80_test to unblock cuda 12.5 build.
This commit is contained in:
Tianlei Wu 2024-07-16 15:58:11 -07:00 committed by GitHub
parent dcc04367b7
commit 760a31c848
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 9 additions and 2 deletions

View file

@ -13,7 +13,7 @@
*/
#pragma once
#if defined(CUDA_VERSION) && CUDA_VERSION <= 12030
#include "test/cuda_host/blkq4_fp16_quant_sm80.h"
#include <random>
@ -197,3 +197,4 @@ void run_blkq4_small_gemm(int m, int n, int k);
} // namespace test
} // namespace cuda
} // namespace onnxruntime
#endif

View file

@ -10,7 +10,7 @@
* This part requires gtest header files, which do not play
* well with CUTLASS headers.
*/
#if defined(CUDA_VERSION) && CUDA_VERSION <= 12030
#include "blkq4_fp16_gemm_sm80.h"
#include "gtest/gtest.h"
@ -341,3 +341,4 @@ TEST(BlkQ4_GEMM, Sm80SmallTileKernelTest) {
} // namespace test
} // namespace onnxruntime
#endif

View file

@ -11,6 +11,9 @@
* well with gtest headers.
*/
// This test has build error with cuda 12.5
#if defined(CUDA_VERSION) && CUDA_VERSION <= 12030
#include "blkq4_fp16_gemm_sm80.h"
#include <random>
@ -532,3 +535,5 @@ template void run_blkq4_small_gemm<128, false, false>(int m, int n, int k);
} // namespace test
} // namespace cuda
} // namespace onnxruntime
#endif