Thank you!
I’ve tried FBGEMM , but it seems it doens’t supported on most android devices (armeabi-v7a. arm64-v8a). Then I found this post FBGEMM with PyTorch Mobile 5 pointing out that FBGEMM is only compatible with x86 achitecture. Do you think I should try FBGEEM
I’ll try out removing  multiple self.add.add in forward