Christian Puhrsch
|
b192e7e415
|
Support non-contiguous NestedTensors for elementwise ops (#87888)
Enables benchmarking of math path of sdp kernel
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87888
Approved by: https://github.com/drisspg
|
2022-10-28 11:26:17 +00:00 |
|
Driss Guessous
|
d06d569e90
|
Update the sdp benchmark to work with nested tensors (#87215)
# Summary
Update the sdp benchmark to work with nested tensors
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87215
Approved by: https://github.com/cpuhrsch
|
2022-10-18 21:38:45 +00:00 |
|
Christian Puhrsch
|
d598290baa
|
Basic SDP benchmark harness (#86729)
Basic benchmark for reference and discussion.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86729
Approved by: https://github.com/drisspg
|
2022-10-12 01:27:59 +00:00 |
|