diff --git a/tests/cpp/operator/test_cast_float8blockwise_grouped.cu b/tests/cpp/operator/test_cast_float8blockwise_grouped.cu index c7d7a73475..bc9f104e17 100644 --- a/tests/cpp/operator/test_cast_float8blockwise_grouped.cu +++ b/tests/cpp/operator/test_cast_float8blockwise_grouped.cu @@ -74,7 +74,8 @@ template void perform_test(ShapeRep shape_rep, BlockDim block_dim, ScalingDir dir, const std::vector& first_dims_h, size_t K, bool force_pow_2_scales, float epsilon) { - if (getDeviceComputeCapability() < hopperComputeCapability) { + if (getDeviceComputeCapability() < hopperComputeCapability || + getDeviceComputeCapability() >= blackwellComputeCapability) { GTEST_SKIP(); }