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

Upsample2d Op Generalization #5621

Open
5 of 9 tasks
mywoodstock opened this issue Feb 24, 2024 · 0 comments
Open
5 of 9 tasks

Upsample2d Op Generalization #5621

mywoodstock opened this issue Feb 24, 2024 · 0 comments
Assignees
Labels
CNNs conv generality feature kernels kernels, such as hlks or llks or below op_cat: window based OPs Op Generalization Generalization and relaxations of requirements in Ops P2 Stable Diffusion

Comments

@mywoodstock
Copy link
Contributor

mywoodstock commented Feb 24, 2024

Upsample Op support:

Layouts

  • ROW_MAJOR
  • TILE

Dataformats

  • BFLOAT16
  • BFP8_B

Sharding

  • HEIGHT_SHARDING
  • BLOCK_SHARDING
  • WIDTH_SHARDING

Remove Restrictions

  • input_nsticks_per_core % in_w == 0
  • input_nsticks % ncores_nhw == 0
@mywoodstock mywoodstock self-assigned this Feb 24, 2024
@mywoodstock mywoodstock added kernels kernels, such as hlks or llks or below P2 CNNs Stable Diffusion Op Generalization Generalization and relaxations of requirements in Ops labels Feb 24, 2024
@mywoodstock mywoodstock assigned nkpatel-tt and unassigned mywoodstock Nov 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CNNs conv generality feature kernels kernels, such as hlks or llks or below op_cat: window based OPs Op Generalization Generalization and relaxations of requirements in Ops P2 Stable Diffusion
Projects
None yet
Development

No branches or pull requests

4 participants