#-------------------------------------------------------------------------------
# Build the mlir bytecode modules with iree-compile. Note the last two flags
# are for RVV support.
#-------------------------------------------------------------------------------

springbok_modules(
  NAME
    mobilenet_v1
  SRC
    "$ENV{ROOTDIR}/ml/ml-models-public/quant_models/mobilenet_v1_0.25_224_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_mobilenet_v1"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    mobilenet_v2
  SRC
    "$ENV{ROOTDIR}/ml/ml-models-public/quant_models/mobilenet_v2_1.0_224_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_mobilenet_v2"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    person_detection
  SRC
    "$ENV{ROOTDIR}/ml/ml-models-public/quant_models/person_detection.tflite"
  C_IDENTIFIER
    "samples_quant_model_person_detection"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

if(${BUILD_INTERNAL_MODELS})

springbok_modules(
  NAME
    barcode
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/barcode_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_barcode"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    daredevil
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/daredevil_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_daredevil"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    fssd_25_8bit_v2
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/fssd_25_8bit_v2_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_fssd_25_8bit_v2"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    scenenet_v2
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/scenenet_v2_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_scenenet_v2"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    semantic_lift
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/semantic_lift_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_semantic_lift"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

springbok_modules(
  NAME
    voice_commands
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/quant_models/voice_commands_quant.tflite"
  C_IDENTIFIER
    "samples_quant_model_voice_commands"
  FLAGS
    "-iree-input-type=tosa"
    "-riscv-v-vector-bits-min=512"
    "-riscv-v-fixed-length-vector-lmul-max=8"
  PUBLIC
)

endif(${BUILD_INTERNAL_MODELS})

#-------------------------------------------------------------------------------
# Binaries to execute the IREE model input
#-------------------------------------------------------------------------------

iree_model_input(
  NAME
    mobilenet_quant_input
  SHAPE
    "1, 224, 224, 3"
  SRC
    "https://storage.googleapis.com/download.tensorflow.org/ \
    example_images/YellowLabradorLooking_new.jpg"
  QUANT
)

iree_model_input(
  NAME
    person_detection_quant_input
  SHAPE
    "1, 96, 96, 1"
  SRC
    "https://github.com/tensorflow/tflite-micro/raw/aeac6f39e5c7475cea20c54e86d41e3a38312546/ \
    tensorflow/lite/micro/examples/person_detection/testdata/person.bmp"
  QUANT
)

if(${BUILD_INTERNAL_MODELS})

iree_model_input(
  NAME
    daredevil_quant_input
  SHAPE
    "1, 15360, 1"
  SRC
    "$ENV{ROOTDIR}/ml/ml-models/test_data/golden_whistle.wav"
  QUANT
)

iree_model_input(
  NAME
    fssd_quant_input
  SHAPE
    "1, 480, 640, 1"
  SRC
    "$ENV{ROOTDIR}/sw/vec_iree/samples/risp4ml/test_data/faces_480x640_uint8_numpy_bayer.bin"
  QUANT
)

endif(${BUILD_INTERNAL_MODELS})

#-------------------------------------------------------------------------------
# Binaries to execute the MLIR bytecode modules
#-------------------------------------------------------------------------------


# If the program requires a larger stack size, add
#
#   LINKOPTS
#     "LINKER:--defsym=__stack_size__=<new stack size>"
#
# to increase it.

iree_cc_binary(
  NAME
    mobilenet_v1_bytecode_static
  SRCS
    "mobilenet_v1.c"
  DEPS
    ::mobilenet_quant_input_c
    ::mobilenet_v1_bytecode_module_static
    ::mobilenet_v1_bytecode_module_static_c
    iree::vm::bytecode_module
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
)

iree_cc_binary(
  NAME
    mobilenet_v1_emitc_static
  SRCS
    "mobilenet_v1.c"
  DEPS
    ::mobilenet_quant_input_c
    ::mobilenet_v1_c_module_static_c
    ::mobilenet_v1_c_module_static_emitc
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
  COPTS
    "-DBUILD_EMITC"
)

# TODO(lundong): Temporarily disable some models due to ITCM overflow
#iree_cc_binary(
#  NAME
#    mobilenet_v2_bytecode_static
#  SRCS
#    "mobilenet_v2.c"
#  DEPS
#    ::mobilenet_quant_input_c
#    ::mobilenet_v2_bytecode_module_static
#    ::mobilenet_v2_bytecode_module_static_c
#    iree::vm::bytecode_module
#    samples::util::util
#  LINKOPTS
#    "LINKER:--defsym=__itcm_length__=1500K"
#    "LINKER:--defsym=__stack_size__=150k"
#)

#iree_cc_binary(
#  NAME
#    mobilenet_v2_emitc_static
#  SRCS
#    "mobilenet_v2.c"
#  DEPS
#    ::mobilenet_quant_input_c
#    ::mobilenet_v2_c_module_static_c
#    ::mobilenet_v2_c_module_static_emitc
#    samples::util::util
#  LINKOPTS
#    "LINKER:--defsym=__itcm_length__=1500K"
#    "LINKER:--defsym=__stack_size__=150k"
#  COPTS
#    "-DBUILD_EMITC"
#)

iree_cc_binary(
  NAME
    person_detection_bytecode_static
  SRCS
    "person_detection.c"
  DEPS
    ::person_detection_bytecode_module_static
    ::person_detection_bytecode_module_static_c
    ::person_detection_quant_input_c
    iree::vm::bytecode_module
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=128k"
)

iree_cc_binary(
  NAME
    person_detection_emitc_static
  SRCS
    "person_detection.c"
  DEPS
    ::person_detection_c_module_static_c
    ::person_detection_c_module_static_emitc
    ::person_detection_quant_input_c
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=128k"
  COPTS
    "-DBUILD_EMITC"
)

if(NOT ${BUILD_INTERNAL_MODELS})
  return()
endif()

iree_cc_binary(
  NAME
    barcode_bytecode_static
  SRCS
    "barcode.c"
  DEPS
    ::barcode_bytecode_module_static
    ::barcode_bytecode_module_static_c
    iree::vm::bytecode_module
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=150k"
)

#iree_cc_binary(
#  NAME
#    barcode_emitc_static
#  SRCS
#    "barcode.c"
#  DEPS
#    ::barcode_c_module_static_c
#    ::barcode_c_module_static_emitc
#    iree::samples::util::util
#  LINKOPTS
#    "LINKER:--defsym=__itcm_length__=1100K"
#    "LINKER:--defsym=__stack_size__=100k"
#  COPTS
#    "-DBUILD_EMITC"
#)

iree_cc_binary(
  NAME
    daredevil_bytecode_static
  SRCS
    "daredevil.c"
  DEPS
    ::daredevil_bytecode_module_static
    ::daredevil_bytecode_module_static_c
    ::daredevil_quant_input_c
    iree::vm::bytecode_module
    samples::audio_prep::mfcc
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
)

iree_cc_binary(
  NAME
    daredevil_emitc_static
  SRCS
    "daredevil.c"
  DEPS
    ::daredevil_c_module_static_c
    ::daredevil_c_module_static_emitc
    ::daredevil_quant_input_c
    samples::audio_prep::mfcc
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
  COPTS
    "-DBUILD_EMITC"
)

iree_cc_binary(
  NAME
    fssd_25_8bit_v2_bytecode_static
  SRCS
    "fssd_25_8bit_v2.c"
  DEPS
    ::fssd_quant_input_c
    ::fssd_25_8bit_v2_bytecode_module_static
    ::fssd_25_8bit_v2_bytecode_module_static_c
    iree::vm::bytecode_module
    samples::risp4ml::pipeline::pipeline
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
)

iree_cc_binary(
  NAME
    fssd_25_8bit_v2_emitc_static
  SRCS
    "fssd_25_8bit_v2.c"
  DEPS
    ::fssd_quant_input_c
    ::fssd_25_8bit_v2_c_module_static_c
    ::fssd_25_8bit_v2_c_module_static_emitc
    samples::risp4ml::pipeline::pipeline
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
  COPTS
    "-DBUILD_EMITC"
)

#iree_cc_binary(
#  NAME
#    scenenet_v2_bytecode_static
#  SRCS
#    "scenenet_v2.c"
#  DEPS
#    ::scenenet_v2_bytecode_module_static
#    ::scenenet_v2_bytecode_module_static_c
#    iree::vm::bytecode_module
#    samples::util::util
#  LINKOPTS
#    "LINKER:--defsym=__itcm_length__=1500K"
#    "LINKER:--defsym=__stack_size__=150k"
#)

#iree_cc_binary(
#  NAME
#    scenenet_v2_emitc_static
#  SRCS
#    "scenenet_v2.c"
#  DEPS
#    ::scenenet_v2_c_module_static_c
#    ::scenenet_v2_c_module_static_emitc
#    samples::util::util
#  LINKOPTS
#    "LINKER:--defsym=__itcm_length__=1500K"
#    "LINKER:--defsym=__stack_size__=150k"
#  COPTS
#    "-DBUILD_EMITC"
#)

iree_cc_binary(
  NAME
    semantic_lift_bytecode_static
  SRCS
    "semantic_lift.c"
  DEPS
    ::semantic_lift_bytecode_module_static
    ::semantic_lift_bytecode_module_static_c
    iree::vm::bytecode_module
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
)

iree_cc_binary(
  NAME
    semantic_lift_emitc_static
  SRCS
    "semantic_lift.c"
  DEPS
    ::semantic_lift_c_module_static_c
    ::semantic_lift_c_module_static_emitc
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__itcm_length__=1M"
    "LINKER:--defsym=__stack_size__=300k"
  COPTS
    "-DBUILD_EMITC"
)

iree_cc_binary(
  NAME
    voice_commands_bytecode_static
  SRCS
    "voice_commands.c"
  DEPS
    ::voice_commands_bytecode_module_static
    ::voice_commands_bytecode_module_static_c
    iree::vm::bytecode_module
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__stack_size__=250k"
)

iree_cc_binary(
  NAME
    voice_commands_emitc_static
  SRCS
    "voice_commands.c"
  DEPS
    ::voice_commands_c_module_static_c
    ::voice_commands_c_module_static_emitc
    samples::util::util
  LINKOPTS
    "LINKER:--defsym=__stack_size__=250k"
  COPTS
    "-DBUILD_EMITC"
)