Skip to content

Commit bc457e1

Browse files
DefTruthlk-chen
authored andcommitted
[Misc] remove warning if triton>=3.2.0 (vllm-project#16553)
Signed-off-by: DefTruth <[email protected]>
1 parent dda7a1f commit bc457e1

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

vllm/attention/ops/triton_decode_attention.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,12 @@
3939

4040
logger = logging.getLogger(__name__)
4141

42-
# TODO: Remove this when triton>=3.2.0. This issue will not affect performance
43-
# and accuracy.
44-
logger.warning(
45-
"The following error message 'operation scheduled before its operands' "
46-
"can be ignored.")
42+
# Only print the following warnings when triton version < 3.2.0.
43+
# The issue won't affect performance or accuracy.
44+
if triton.__version__ < '3.2.0':
45+
logger.warning(
46+
"The following error message 'operation scheduled before its operands' "
47+
"can be ignored.")
4748

4849

4950
@triton.jit

0 commit comments

Comments
 (0)