From 47d2b9754caeb3929fc668cc1d99e87c026081df Mon Sep 17 00:00:00 2001 From: TNTwise Date: Sun, 8 Sep 2024 22:58:50 +0000 Subject: [PATCH] change engine gen name --- backend/src/InterpolateTorch.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/src/InterpolateTorch.py b/backend/src/InterpolateTorch.py index 291fc356..00b217af 100644 --- a/backend/src/InterpolateTorch.py +++ b/backend/src/InterpolateTorch.py @@ -11,7 +11,7 @@ check_bfloat16_support, log ) - +from .version import __version__ torch.set_float32_matmul_precision("high") torch.set_grad_enabled(False) @@ -299,6 +299,7 @@ def __init__( + f"_ensemble-{ensemble}" + f"_{torch.cuda.get_device_name(self.device)}" + f"torch_tensorrt-{torch_tensorrt.__version__}" + + f"backend_version-{__version__}" + f"_trt-{tensorrt.__version__}" + ( f"_workspace-{trt_workspace_size}"