change engine gen name

TNTwise · Sep 8, 2024 · 47d2b97 · 47d2b97
1 parent 8cf5173
commit 47d2b97
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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}"