UMA & from_keras: putting it all together

Hi @slai-nick

I got your test to work by adjusting it according to the mobilenet test: https://github.com/apache/tvm/blob/main/tests/python/contrib/test_uma/test_uma_pipeline.py#L111

  • pass the params to the AOTModel
  • set the target to [target_c, target]

The last remaining errors you are seeing seem to be related to your target architecture and compiler. Probably, AOT is not yet supported for the M1.

1 Like