onnx-mlir
onnx-mlir copied to clipboard
Show the compilation phases
This patch is to show the compilation phases which is helpful when compiling a big model that takes a long time.
This is one example of the printed information:
$ onnx-mlir -mcpu=z16 -maccel=NNPA bert_base_uncased_1.onnx
[1/5] Wed Jul 3 03:36:50 2024 Importing ONNX Model to MLIR Module
[2/5] Wed Jul 3 03:36:51 2024 Compiling and Optimizing MLIR Module
[3/5] Wed Jul 3 03:37:14 2024 Translating MLIR Module to LLVM and Generating LLVM Optimized Bitcode
[4/5] Wed Jul 3 03:37:35 2024 Generating Object from LLVM Bitcode
[5/5] Wed Jul 3 03:37:39 2024 Linking and Generating the Output Shared Library
Starting time is added to each phase so that users know how long the current phase takes
FYI @tehbone
Is this default behavior or enabled via switch?
Is this default behavior or enabled via switch?
Yes, it's on by default. Let me know if you want a switch or set it off.
Jenkins Linux s390x Build #15079 [push] Show the compilation pha... started at 20:37
Jenkins Linux ppc64le Build #14104 [push] Show the compilation pha... started at 20:47
Jenkins Linux amd64 Build #15074 [push] Show the compilation pha... started at 19:37
Jenkins Linux amd64 Build #15074 [push] Show the compilation pha... passed after 1 hr 6 min
Jenkins Linux s390x Build #15079 [push] Show the compilation pha... passed after 1 hr 33 min
Jenkins Linux ppc64le Build #14104 [push] Show the compilation pha... passed after 2 hr 7 min