wanglei91 commented on PR #10882: URL: https://github.com/apache/tvm/pull/10882#issuecomment-1537097325
@apivovarov Thanks, I use this patch to test SD unet, it indeed support large >2GB models(mold instead of ld linker could also support it), but both this patch and mold linker still truncate when more than 3.1G(for may case, this patch produces a 3.1G unet.so, save more tensors than mold), I don't know if somebody else also meet this issue... print when save: header: 15951258332257624383 cpu_dev.device_type: 1 tensor->ndim: 3 shap0: 16 shape1: 4096 shape2: 77 -------------------------------------------------------------------- header: 15951258332257624383 cpu_dev.device_type: 1 tensor->ndim: 3 shap0: 640 shape1: 1 shape2: 1 print when load: header: 15951258332257624383 shap0: 16 shape1: 4096 shape2: 77 -------------------------------------------------------------------- header: 160445370461692650 include/tvm/runtime/ndarray.h", line 512 TVMError: Check failed: (header == kTVMNDArrayMagic) is false: Invalid DLTensor file format -- 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]
