I compiled torch with vulkan backend, and saved the model with vulkan backend. However, when I do the inference on android device, it throwed the following error:
com.facebook.jni.CppException: Could not run ‘vulkan_prepack::run_conv2d_context’ with arguments from the ‘CPU’ backend.
Please let me know what’s the cause of this issue. Thanks