[https://nvbugs/5702795][fix] Remove the warning message for aten.log. (#9665)

nv-guomingz · web-flow · commit b5e2b9b51fd8 · 2025-12-04T00:02:15.000+08:00
Signed-off-by: nv-guomingz &lt;137257613+nv-guomingz@users.noreply.github.com&gt;
diff --git 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
     }
 

Original file line number	Diff line number	Diff line change
`@@ -71,6 +71,7 @@ class MetaInitMode(TorchDispatchMode):`
`71`	`71`	`random_init_ops = {`
`72`	`72`	`aten.normal_.default,`
`73`	`73`	`aten.uniform_.default,`
	`74`	`+ aten.log.default,`
`74`	`75`	`# TODO: this is not a exhaustive list for random init ops, add as needed`
`75`	`76`	`}`
`76`	`77`