Skip to content

Commit 1d4bc03

Browse files
committed
fix condition.
1 parent f72b0cb commit 1d4bc03

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_modeling_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1417,7 +1417,7 @@ def get_memory_usage(storage_dtype, compute_dtype):
14171417
self.assertTrue(fp8_e4m3_bf16_memory_footprint < fp8_e4m3_fp32_memory_footprint < fp32_memory_footprint)
14181418
# NOTE: the following assertion will fail on our CI (running Tesla T4) due to bf16 using more memory than fp32.
14191419
# On other devices, such as DGX (Ampere) and Audace (Ada), the test passes.
1420-
if compute_capability < 8.9:
1420+
if compute_capability >= 8.9:
14211421
self.assertTrue(fp8_e4m3_bf16_max_memory < fp8_e4m3_fp32_max_memory)
14221422
# On this dummy test case with a small model, sometimes fp8_e4m3_fp32 max memory usage is higher than fp32 by a few
14231423
# bytes. This only happens for some models, so we allow a small tolerance.

0 commit comments

Comments
 (0)