[Debug] Failed to export library with LLVM on ARM CPU when using PReLU

Questions
#1

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:

tvm_llvm_error
tvm_llvm_error.png1073×959 148 KB

If I close the TOPI schedule for conv2d or change PReLU to ReLU, it works fine. Have you met the problem before?

1 Like
#2

Same problem. Relu is ok but PRelu failed on rk3399.

#3

Does this happen on x86? Or is it ARM only issue?

#4

Arm only. x86 works well.

#5

yes. pReLu will cause error when compiling on arm-cpu but works for mali-gpu.

#6

https://github.com/dmlc/tvm/pull/2381 should resolve your issue. Reply it here in case of you mute github notification.

1 Like