-
Notifications
You must be signed in to change notification settings - Fork 86
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
[WIP] IRON API and Dev Tools #1732
base: main
Are you sure you want to change the base?
Conversation
2e19d04
to
ea55152
Compare
Coverage ReportCreated: 2024-10-14 18:21Click here for information about interpreting this report.
Generated by llvm-cov -- llvm version 14.0.0 |
229acf4
to
e103e35
Compare
4251bea
to
1c22e32
Compare
f83c852
to
649065c
Compare
7fda896
to
181368a
Compare
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TILE_SIZE = TILE_WIDTH * TILE_HEIGHT | ||
|
||
with mlir_mod_ctx() as ctx: | ||
def my_matrix_add_one(): |
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.
[black] reported by reviewdog 🐶
def my_matrix_add_one(): | |
def my_matrix_add_one(): |
|
||
@runtime_sequence(tile_ty, tile_ty, tile_ty) | ||
def sequence(inTensor, _, outTensor): | ||
npu_dma_memcpy_nd(metadata=of_in1, bd_id=1, |
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.
[black] reported by reviewdog 🐶
npu_dma_memcpy_nd(metadata=of_in1, bd_id=1, | |
npu_dma_memcpy_nd( | |
metadata=of_in1, | |
bd_id=1, |
issue_token=True, | ||
) | ||
|
||
npu_dma_memcpy_nd(metadata=of_out1, bd_id=0, |
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.
[black] reported by reviewdog 🐶
npu_dma_memcpy_nd(metadata=of_out1, bd_id=0, | |
npu_dma_memcpy_nd( | |
metadata=of_out1, | |
bd_id=0, |
TILE_HEIGHT, TILE_WIDTH = 8, 16 | ||
TILE_SIZE = TILE_WIDTH * TILE_HEIGHT | ||
|
||
def my_matrix_add_one(): |
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.
[black] reported by reviewdog 🐶
def my_matrix_add_one(): | |
def my_matrix_add_one(): |
my_worker = Worker(core_fn, | ||
fn_args=[of_in.second, of_out.first], | ||
while_true=True) |
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.
[black] reported by reviewdog 🐶
my_worker = Worker(core_fn, | |
fn_args=[of_in.second, of_out.first], | |
while_true=True) | |
my_worker = Worker(core_fn, fn_args=[of_in.second, of_out.first], while_true=True) |
Please ignore, not nearly ready for consideration.