-
Notifications
You must be signed in to change notification settings - Fork 13
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ttnn ops backend metal wrapper lib #1230
ttnn ops backend metal wrapper lib #1230
Conversation
467b4b7
to
c5b33f2
Compare
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.
Hey @mbezuljTT, these changes look great! Some comments inline
eeb084a
to
6577759
Compare
bbc53b4
to
0634367
Compare
fddfb02
to
8bc11da
Compare
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.
Clang-Tidy
found issue(s) with the introduced code (1/1)
8bc11da
to
6cba062
Compare
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.
Clang-Tidy
found issue(s) with the introduced code (1/1)
217998a
to
d1d1019
Compare
e17d8c8
to
4028b7b
Compare
Addressed all comments, merging this PR. afterthoughts are OK, please reach out; |
TTNNOpModelLib is to be used for op model interface (constraints, l1, perf)
TTNNOpModelLib is static library that allows us to call ttnn and tt-metal API from mlir TTNN Dialect to enable direct calls to compiler ops API to query for arguments validity, L1 usage, and performance.
The specifics are LLVM/MLIR have exceptions disabled and are built with C++ 17.
Metal is built with exceptions and with C++ 20.
As we are building MLIR and metal using same compiler version, we shouldnt get any ABI issues.
Exceptions would be handled within TTNNWrapperLib.
Added
TTMLIR_ENABLE_OPMODEL
to be used for this specific build, build has been added to CI. No tests are added, as we add MNIST ops, we shall add MNIST optimizer test. We shall agree when to make this mode enabled by default.@nsmithtt @nobradovictt @sdjordjevicTT I think this should be good initial version
@vmilosevic please take a look at CI changes,