refactors
This commit is contained in:
@@ -145,4 +145,24 @@ raptor_apply_patch(
|
||||
"Skip output emission for PIM accelerator"
|
||||
)
|
||||
|
||||
# Count the PIM status messages instead of the unused standard backend phases
|
||||
raptor_apply_patch(
|
||||
"${ONNX_MLIR_DIR}/src/Compiler/CompilerUtils.cpp"
|
||||
"void showCompilePhase(std::string msg) {"
|
||||
[=[void showCompilePhase(std::string msg) {
|
||||
if (llvm::is_contained(maccel, accel::Accelerator::Kind::PIM)) {
|
||||
if (pimOnlyCodegen)
|
||||
TOTAL_COMPILE_PHASE = 2;
|
||||
else if (pimEmissionTarget == EmitSpatial)
|
||||
TOTAL_COMPILE_PHASE = 3;
|
||||
else if (pimEmissionTarget == EmitPim)
|
||||
TOTAL_COMPILE_PHASE = 4;
|
||||
else if (pimEmissionTarget == EmitPimBufferized)
|
||||
TOTAL_COMPILE_PHASE = 5;
|
||||
else
|
||||
TOTAL_COMPILE_PHASE = 9;
|
||||
}]=]
|
||||
"Use PIM compile phase count"
|
||||
)
|
||||
|
||||
add_subdirectory(onnx-mlir)
|
||||
|
||||
Reference in New Issue
Block a user