export onnx error ( We don't have an op for aten::mul but it isn't a special case )

[reskipper]

when i export GeMap to onnx ，report error as below : RuntimeError: 0INTERNAL ASSERT FAILED at "../torch/csrc/jit/ir/alias_analysis.cpp":532, please report a bug to PyTorch. We don't have an op for aten::mul but it isn't a special case. Argument types: Tensor, bool,

I also tried PyTorch 1.10.0, and encountered the same error. Any ideas on how to tackle this? Thanks a bunch!