We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 723c08c commit 0643368Copy full SHA for 0643368
tripy/tripy/constants.py
@@ -16,7 +16,10 @@
16
#
17
18
CONSTANT_IR_PRINT_VOLUME_THRESHOLD = 5
19
-STROGE_OP_CACHE_VOLUME_THRESHOLD = 64 # TensorRT dim supports up to 8 dimensions
+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
23
24
"""
25
A volume threshold for displaying constants in IR logging messages.
0 commit comments