-
Notifications
You must be signed in to change notification settings - Fork 7
Move bottom up fuser declaration to header file #23
Conversation
BUILD
Outdated
@@ -18,6 +18,7 @@ td_library( | |||
"include/mlir-tcp/Dialect/IR/TcpEnums.td", | |||
"include/mlir-tcp/Dialect/IR/TcpOps.td", | |||
"include/mlir-tcp/Dialect/IR/TcpTypes.td", | |||
"include/mlir-tcp/Dialect/IR/TcpOpsCruiseInternal.td", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
unintentional?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for catching this. Fixed
BUILD
Outdated
"lib/Conversion/TorchToTcp/TorchToTcpCruiseInternal.cpp", | ||
"lib/Conversion/TorchToTcp/CruiseInternalPatterns.cpp", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please revert
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed
std::function<void(Operation *, PatternRewriter &rewriter)>; | ||
|
||
GenericBottomUpFuser(MLIRContext *context, CanFuseFuncType canFuseCallback, | ||
PostProcessingFuncType postFuncCallback = nullptr) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please add brief docstrings/comments on what the constructor args (canFuseCallback
and postFuncCallback
) are meant to do. Especially since the upstream passes (like TcpFuseElementwiseOpsPass
) don't use the postFuncCallback
, it is unclear from the source code in the header.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks and LGTM! Please address the docstring comment before landing.
Move bottom up fuser declaration to header file No testing required since it's a minor restructuring --------- Co-authored-by: Muhammad Abubakar <muhammad.abubakar@getcruise.com>
Add missing file from previous PR
Move bottom up fuser declaration to header file
No testing required since it's a minor restructuring