Skip to content

Commit 4b60632

Browse files
committed
remove insufficent mem check and skip
1 parent f0d1caf commit 4b60632

1 file changed

Lines changed: 0 additions & 11 deletions

File tree

cpp/tests/linalg/gemm_large.cpp

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -28,21 +28,10 @@ TEST(Raft, GemmCublasLt136LargeSpan)
2828
constexpr std::array<std::int64_t, 3> sample_counts{134217727, 134217728, 134217729};
2929
constexpr auto max_samples = sample_counts.back();
3030
constexpr auto a_elements = static_cast<std::size_t>(max_samples) * lda;
31-
constexpr auto required_bytes =
32-
(a_elements + informative + static_cast<std::size_t>(max_samples)) * sizeof(float);
33-
constexpr std::size_t allocation_headroom = std::size_t{1} << 30;
3431

3532
raft::resources resources;
3633
const auto stream = raft::resource::get_cuda_stream(resources);
3734

38-
std::size_t free_bytes{};
39-
std::size_t total_bytes{};
40-
RAFT_CUDA_TRY(cudaMemGetInfo(&free_bytes, &total_bytes));
41-
if (free_bytes < required_bytes + allocation_headroom) {
42-
GTEST_SKIP() << "This regression test needs " << required_bytes + allocation_headroom
43-
<< " free device bytes, but only " << free_bytes << " are available";
44-
}
45-
4635
rmm::device_uvector<float> a(a_elements, stream);
4736
rmm::device_uvector<float> b(informative, stream);
4837
rmm::device_uvector<float> c(max_samples, stream);

0 commit comments

Comments
 (0)