Skip to content

Commit 0643368

Browse files
committed
update comment
1 parent 723c08c commit 0643368

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tripy/tripy/constants.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@
1616
#
1717

1818
CONSTANT_IR_PRINT_VOLUME_THRESHOLD = 5
19-
STROGE_OP_CACHE_VOLUME_THRESHOLD = 64 # TensorRT dim supports up to 8 dimensions
19+
STROGE_OP_CACHE_VOLUME_THRESHOLD = (
20+
64 # We want to exclude shape tensors, and TensorRT shape dims supports up to 8 dimensions, so
21+
)
22+
# we want a threshold value a lot bigger than 8
2023

2124
"""
2225
A volume threshold for displaying constants in IR logging messages.

0 commit comments

Comments
 (0)