This is an automated email from the ASF dual-hosted git repository.
tlopex pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new c043e4fc8d [Relax] Add NN operator attributes include to TensorRT
codegen (#18690)
c043e4fc8d is described below
commit c043e4fc8d5c62f7073be69cf57f8f383634936f
Author: Guan-Ming (Wesley) Chiu <[email protected]>
AuthorDate: Thu Jan 29 04:59:41 2026 +0800
[Relax] Add NN operator attributes include to TensorRT codegen (#18690)
## Why
The header now exists and provides operator attribute definitions needed
for TensorRT codegen.
## How
Add #include <tvm/relax/attrs/nn.h> to TensorRT codegen
---
src/relax/backend/contrib/tensorrt/codegen.cc | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/src/relax/backend/contrib/tensorrt/codegen.cc
b/src/relax/backend/contrib/tensorrt/codegen.cc
index 0adeb2d475..02d9a61a80 100644
--- a/src/relax/backend/contrib/tensorrt/codegen.cc
+++ b/src/relax/backend/contrib/tensorrt/codegen.cc
@@ -24,8 +24,7 @@
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/module.h>
#include <tvm/ir/transform.h>
-// TODO(sunggg): add operator attribute when it's ready
-// #include <tvm/relax/attrs/nn.h>
+#include <tvm/relax/attrs/nn.h>
#include <tvm/relax/type.h>
#include <memory>
