search

Tencent / TPAT

TensorRT Plugin Autogen Tool
Apache License 2.0
365 stars 42 forks source link

RandomNormal not supported for frontend ONNX #19

Open sunkenQ opened 1 year ago

sunkenQ commented 1 year ago

I want create a RandomNormal op and I find this operator marked "Y" in TPAT-1.0 Operator Schemas. I build BlazerML-TVM successfully and use command line python onnx_to_plugin.py -i randn_test.onnx -o output.onnx -t RandomNormal, there was an ERROR tvm.error.OpNotImplemented: The following operators are not supported for frontend ONNX: RandomNormal. In /mypath/TPAT/3rdparty/blazerml-tvm/python/tvm/relay/frontend/onnx.py, I find RandomNormal operator was commented in function _get_convert_map, I think it means RandomNormal was not supported. So how TPAT-1.0 create .so file for RandomNormal ?

buptqq commented 1 year ago

you can instead subodule('tpat' branch of BlazerML-tvm) of the 'dev' branch of BlazerML-tvm：https://github.com/Tencent/BlazerML-tvm/tree/dev The ‘dev’ branch of BlazerML-TVM has added multiple ops support：‘Scan’, ‘RandomNormal’, 'RandomNormalLike', 'RandomUniformLike'.