std::unique_ptr<Pass> pass;
- if (cli::target == NNC_TARGET_X86_CPP) {
+ if (cli::target == NNC_TARGET_ARM_CPP) {
+ pass = std::move(std::unique_ptr<Pass>(new CPPCodeGenerator()));
+ } else if (cli::target == NNC_TARGET_X86_CPP) {
pass = std::move(std::unique_ptr<Pass>(new CPPCodeGenerator()));
} else if (cli::target == NNC_TARGET_ARM_GPU_CPP) {
pass = std::move(std::unique_ptr<Pass>(new AclCppCodeGenerator()));
);
Option<std::string> target(optname("--target"),
overview("select target language to emit for given architecture."
- "Valid values are '" NNC_TARGET_X86_CPP "', '" NNC_TARGET_ARM_GPU_CPP "', '" NNC_TARGET_INTERPRETER "'"),
+ "Valid values are '"
+ NNC_TARGET_ARM_CPP "', '"
+ NNC_TARGET_X86_CPP "', '"
+ NNC_TARGET_ARM_GPU_CPP "', '"
+ NNC_TARGET_INTERPRETER "'"),
std::string(),
optional(false),
- optvalues(NNC_TARGET_X86_CPP "," NNC_TARGET_ARM_GPU_CPP "," NNC_TARGET_INTERPRETER),
+ optvalues(NNC_TARGET_ARM_CPP ","
+ NNC_TARGET_X86_CPP ","
+ NNC_TARGET_ARM_GPU_CPP ","
+ NNC_TARGET_INTERPRETER),
nullptr,
separators("="));
extern Option<bool> onnxFrontend; // frontend for ONNX AI framework
// valid values for target option
+#define NNC_TARGET_ARM_CPP "arm-c++"
#define NNC_TARGET_X86_CPP "x86-c++"
#define NNC_TARGET_ARM_GPU_CPP "arm-gpu-c++"
#define NNC_TARGET_INTERPRETER "interpreter"