I tried to compile models with PReLU for ARM CPU on Firefly-RK3399.
target = tvm.target.create(‘llvm -device=arm_cpu -model=rk3399 -target=aarch64-linux-gnu’)
Errors when using lib.export_library to generate code with LLVM:
If I close the TOPI schedule for conv2d or change PReLU to ReLU, it works fine. Have you met the problem before?
1 Like
nttstar
2
Same problem. Relu is ok but PRelu failed on rk3399.
masahi
3
Does this happen on x86? Or is it ARM only issue?
nttstar
4
Arm only. x86 works well.
jackyu
5
yes. pReLu will cause error when compiling on arm-cpu but works for mali-gpu.
https://github.com/dmlc/tvm/pull/2381 should resolve your issue. Reply it here in case of you mute github notification.
1 Like
