We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f4d2185 commit 19dfc40Copy full SHA for 19dfc40
1 file changed
modelopt/onnx/quantization/autotune/__main__.py
@@ -17,6 +17,7 @@
17
18
import argparse
19
import sys
20
+import tempfile
21
from pathlib import Path
22
23
from modelopt.onnx.logging_config import logger
@@ -29,7 +30,7 @@
29
30
DEFAULT_NUM_SCHEMES = 30
31
DEFAULT_QUANT_TYPE = "int8"
32
DEFAULT_DQ_DTYPE = "float32"
-DEFAULT_TIMING_CACHE = "/tmp/trtexec_timing.cache" # nosec B108
33
+DEFAULT_TIMING_CACHE = str(Path(tempfile.gettempdir()) / "trtexec_timing.cache")
34
DEFAULT_WARMUP_RUNS = 5
35
DEFAULT_TIMING_RUNS = 20
36
0 commit comments