Mousius commented on issue #8717: URL: https://github.com/apache/tvm/issues/8717#issuecomment-910186908
Apologies for the slow reply, I've just checked out `main` fresh and checked this still happens. To recreate this I took the AOT test `test_quant_mobilenet_tfl` which uses `mobilenet_v1_1.0_224_quant.tflite` from https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224_quant.tgz. I put a breakpoint just after the code is generated (here-ish: https://github.com/apache/tvm/blob/main/tests/python/relay/aot/aot_test_utils.py#L570) , and you can see large allocations in `tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_clip_cast_cast_21` within `default_lib1.c` in the temporary directory which is named in the variable `base_path`: ``` void* PaddedInput = TVMBackendAllocWorkspace(1, dev_id, (uint64_t)861184, 0, 16); if (PaddedInput == NULL) { return -1; } void* DepthwiseConv2d = TVMBackendAllocWorkspace(1, dev_id, (uint64_t)1605632, 0, 32); if (DepthwiseConv2d == NULL) { return -1; } ``` Hope that helps :smile_cat: -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
