From 792aa3b3261f677b2be4eddb1a210c9190fe87e8 Mon Sep 17 00:00:00 2001 From: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com> Date: Wed, 3 Dec 2025 11:46:14 +0000 Subject: [PATCH] [https://nvbugs/5702795][fix] Remove the warning message for aten.log. Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com> --- tensorrt_llm/_torch/models/modeling_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorrt_llm/_torch/models/modeling_utils.py b/tensorrt_llm/_torch/models/modeling_utils.py index 22d544f92b1b..c4e7906bbdb8 100755 --- a/tensorrt_llm/_torch/models/modeling_utils.py +++ b/tensorrt_llm/_torch/models/modeling_utils.py @@ -71,6 +71,7 @@ class MetaInitMode(TorchDispatchMode): random_init_ops = { aten.normal_.default, aten.uniform_.default, + aten.log.default, # TODO: this is not a exhaustive list for random init ops, add as needed }