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, ...@@ -33,8 +33,8 @@ void performTest(const size_t N, const size_t H, const bool zero_centered_gamma,
return; return;
} }
if (getDeviceComputeCapability() < blackwellComputeCapability && use_cudnn) { if (getDeviceComputeCapability() < hopperComputeCapability && use_cudnn) {
GTEST_SKIP() << "cuDNN normalizations not supported on pre-Blackwell GPUs yet!"; GTEST_SKIP() << "cuDNN normalizations not supported on pre-Hopper GPUs yet!";
} }
using WeightType = InputType; 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