Add custom attributes to Node in JIT

1

Is there any way to add custom attributes to JIT Node ? I need to use this attribute when I run custom passes on JIT Graph.

2 Likes
2

Hi, did you find the solution?
I tried register_buffer and register_parameter APIs (they are on module level but not Node, but they fit my need), but neither works. I got below error when calling them for a loaded JIT model.
"
RuntimeError: Can’t add a new parameter after ScriptModule construction.
"
And below error when calling a new ScriptModule.
"
AttributeError: cannot assign buffer before Module.init() call
"
any input will be appreciated.

thanks,