Unverified Commit 86c50977 authored by Phuong Nguyen's avatar Phuong Nguyen Committed by GitHub
Browse files

[Test] Enable cuDNN Norm tests in the CPP suite (#1957)



* enable cudnn norm tests
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>

* exclude tests on pre-Hopper
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>
parent 2d4644b7
......@@ -33,8 +33,8 @@ void performTest(const size_t N, const size_t H, const bool zero_centered_gamma,
return;
}
if (getDeviceComputeCapability() < blackwellComputeCapability && use_cudnn) {
GTEST_SKIP() << "cuDNN normalizations not supported on pre-Blackwell GPUs yet!";
if (getDeviceComputeCapability() < hopperComputeCapability && use_cudnn) {
GTEST_SKIP() << "cuDNN normalizations not supported on pre-Hopper GPUs yet!";
}
using WeightType = InputType;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment