Skip to content

support shape broadcast in UOp.alu#15442

Merged
chenyuxyz merged 1 commit intotinygrad:masterfrom
chenyuxyz:uop-alu-broadcast
Mar 24, 2026
Merged

support shape broadcast in UOp.alu#15442
chenyuxyz merged 1 commit intotinygrad:masterfrom
chenyuxyz:uop-alu-broadcast

Conversation

@chenyuxyz
Copy link
Copy Markdown
Collaborator

i think it can integrate tighter, but now Tensor also does ufix from UOp and implicit dtype upcast

@chenyuxyz chenyuxyz marked this pull request as draft March 24, 2026 03:44
@github-actions
Copy link
Copy Markdown
Contributor

Changes

Name                          Lines    Diff    Tokens/Line    Diff
--------------------------  -------  ------  -------------  ------
tinygrad/tensor.py             1443      -2           21.2    -0.0
tinygrad/uop/ops.py            1332      +9           18.3    -0.0
tinygrad/mixin/movement.py      152      -3           19.6    -0.0


total lines changes: +4

i think it can integrate tighter, but now Tensor also does ufix from UOp and implicit dtype upcast
@chenyuxyz chenyuxyz marked this pull request as ready for review March 24, 2026 14:14
@chenyuxyz chenyuxyz merged commit b796084 into tinygrad:master Mar 24, 2026
98 of 99 checks passed
@chenyuxyz chenyuxyz deleted the uop-alu-broadcast branch March 24, 2026 14:15
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.

1 participant