kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>
This commit is contained in:
Guoming Zhang 2025-12-04 00:02:15 +08:00 committed by GitHub
parent 09beaa5933
commit b5e2b9b51f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorrt_llm/_torch/models/modeling_utils.py
View File

@ -71,6 +71,7 @@ class MetaInitMode(TorchDispatchMode):
random_init_ops = { random_init_ops = {
aten.normal_.default, aten.normal_.default,
aten.uniform_.default, aten.uniform_.default,
aten.log.default,
# TODO: this is not a exhaustive list for random init ops, add as needed # TODO: this is not a exhaustive list for random init ops, add as needed
} }