Skip to content

Commit f45adb0

Browse files
fix: correct AmperePeakGflopsCalculation expected value
The test expected 19481.0 GFLOPS but the correct calculation is 108 * 64 * 2 * 1.41 = 19491.84 GFLOPS. Fix the expected value and remove the stale * 1000 from the comment. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
1 parent eb10bc8 commit f45adb0

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/test_device_info_cpu.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,10 +106,10 @@ TEST_F(FakeDeviceProvider, VoltaPeakGflopsCalculation) {
106106
}
107107

108108
TEST_F(FakeDeviceProvider, AmperePeakGflopsCalculation) {
109-
// A100: 108 SMs * 64 cores/SM * 2 (FMA) * 1.41 GHz * 1000
110-
// Expected: ~19.481 TFLOPS
109+
// A100: 108 SMs * 64 cores/SM * 2 (FMA) * 1.41 GHz
110+
// Expected: ~19.492 TFLOPS
111111
float peak_gflops = getTheoreticalPeakGflops(ampere_provider_);
112-
EXPECT_NEAR(peak_gflops, 19481.0f, 0.1f);
112+
EXPECT_NEAR(peak_gflops, 19491.84f, 0.1f);
113113
}
114114

115115
TEST_F(FakeDeviceProvider, VoltaPeakBandwidthCalculation) {

0 commit comments

Comments
 (0)