mikaylagawarecki

Add support for indices to segment_reduce (CPU-only)

2

Stack from [ghstack](https://github.com/ezyang/ghstack): * __->__ #78908 * #78918 * #78907 * #77061

Enable codegen of per-dispatch key derivative formulas in derivatives.yaml

1

`derivatives.yaml` can now take a `dispatch` entry which registers per-autograd dispatch key derivatives such as ``` name: foo(Tensor self, Tensor y) -> Tensor dispatch: default: x: grad y: grad.expand(y.sizes()) AutogradNestedTensor:...

[DO NOT REVIEW] Testing integration

1

[wip] Proof of concept that torch ops in nightly will pass tests + identify gaps in what's in core Gaps with no workarounds (fixes are wip): - [x] integer support...

Fix idx overflows in scatter_cuda kernel to enable benchmark/scatter_segment.py with long rows

1

This PR exists to show the "state of the world" for the "before" benchmark ([**rendered benchmarking results cuda**](https://github.com/rusty1s/pytorch_scatter/blob/58bd3fba7b6cd7564d24339134d41ad18d84ef04/benchmark_before_cuda.md)) - fixes CUDA illegal memory accesses run into by benchmarking script mentioned...

Use fallback approach for nested matmul

2

Stack from [ghstack](https://github.com/ezyang/ghstack): * __->__ #85311

Re-introduce `torch.Tensor.to_padded_tensor`

8

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #85293 Differential Revision: [D39629004](https://our.internmc.facebook.com/intern/diff/D39629004)

Hey `miniz` maintainers, I am wondering whether it would be feasible for the `mz_zip_writer_add_mem_ex_v2` API to only write the metadata associated with the file without requiring `pBuf` to be defined....

[easy] remove some unecessary windows skips for mmap tests

5

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #124891 * #124889

Add MAP_SHARED option for torch.load(mmap=True)

8

Fixes #124528 Going over the options for our MapAllocator and what they do, I don't think any other of them need to be piped up to `torch.load` https://github.com/pytorch/pytorch/blob/4f29103749c5011529f1abb10b1508a682588909/aten/src/ATen/MapAllocator.h#L8-L16 ~However, I...

Add `write_record_metadata` to PyTorchFileWriter

1

Add `PyTorchFileWriter.write_record_metadata(record_name, num_bytes)` that - writes the zipfile header/end of central directory metadata for an entry* - reserves `num_bytes` in the zipfile for the payload. *Since the payload is not...