Skip to content
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

runtime support for full op #644

Merged
merged 1 commit into from
Sep 10, 2024
Merged

runtime support for full op #644

merged 1 commit into from
Sep 10, 2024

Conversation

jnie-TT
Copy link
Contributor

@jnie-TT jnie-TT commented Sep 6, 2024

closes #626

auto shape = ::ttnn::Shape(::tt::tt_metal::Shape(
utils::toShapeFromFBShape(*op->out()->desc()->shape())));
float fillValue = op->fill_value();
// TODO(bug #272), determine correct layout by tile shape in the future
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

For now we can probably assert that there are only 2 tile shapes and therefore infer, [1, 1] -> row major, or [32, 32] -> tiled

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Confirmed with @nsmithtt @nobradovictt that currently we don't have compiler support for dynamically generating tile shapes, will leave this hardcoded and TODO for now until we bring up the support.

@jnie-TT jnie-TT force-pushed the jnie/full_op branch 2 times, most recently from c9f9d56 to 0401c3a Compare September 10, 2024 18:22
@jnie-TT jnie-TT merged commit 1c033bf into main Sep 10, 2024
11 checks passed
uazizTT pushed a commit that referenced this pull request Sep 12, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Implement full op
2 participants