Skip to content

[Type] Tensor 7: Test layout= + needs_grad= grad-buffer round-…#527

Draft
hughperkins wants to merge 2 commits intohp/tensor-stork-6from
hp/tensor-stork-7
Draft

[Type] Tensor 7: Test layout= + needs_grad= grad-buffer round-…#527
hughperkins wants to merge 2 commits intohp/tensor-stork-6from
hp/tensor-stork-7

Conversation

@hughperkins
Copy link
Copy Markdown
Collaborator

…trip

Test-only PR closing the gap left by PRs 5 and 6: the combination of layout= and needs_grad= was checked at allocation time, but not yet exercised through a kernel write/read on the .grad buffer with a non- identity layout.

This pins down the pre-impl POC Q3b finding (grad SNode inherits the primal's axis_seq) so any upstream Quadrants regression in this area surfaces immediately.

Adds 8 tests:

  • rank-2 transposed-storage primal+grad kernel round-trip
  • rank-3 (2, 0, 1) primal+grad kernel round-trip
  • all 6 rank-3 layout permutations with primal+grad written and read

All tests pass; no production-code or doc changes.

Issue: #

Brief Summary

copilot:summary

Walkthrough

copilot:walkthrough

…trip

Test-only PR closing the gap left by PRs 5 and 6: the combination of
layout= and needs_grad= was checked at allocation time, but not yet
exercised through a kernel write/read on the .grad buffer with a non-
identity layout.

This pins down the pre-impl POC Q3b finding (grad SNode inherits the
primal's axis_seq) so any upstream Quadrants regression in this area
surfaces immediately.

Adds 8 tests:
- rank-2 transposed-storage primal+grad kernel round-trip
- rank-3 (2, 0, 1) primal+grad kernel round-trip
- all 6 rank-3 layout permutations with primal+grad written and read

All tests pass; no production-code or doc changes.
@hughperkins
Copy link
Copy Markdown
Collaborator Author

Would prefer to add the ndarray layout first really, but ...

Drops the 'flexible' prefix from filenames and identifiers introduced
in this branch series so the user-visible names are simply 'tensor'.
Also strips PR-N back-references that will be meaningless once these
PRs land. Touches only files owned by this series (no changes to
external/ or unrelated tests).
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