I am working on the implementation of GELU lowering. I am having a problem that when running torch_ods_gen.py
, aten::gelu
is not a registered op.
get_registered_ops()
in _jit_ir_importer.cpython-38-x86_64-linux-gnu.so
does not contain it.
Does anyone know what I am missing?
It turned out that GELU does not have a in-place variant.