pytorch
c771d734 - [composite compliance] fix max_pool1d (#84127)

Commit
2 years ago
[composite compliance] fix max_pool1d (#84127) max_pool1d has a fast path for CPU tensors that do not require grad that directly accesses the data_ptr. This PR makes the change that if the input Tensor is a Tensor Subclass, then we want to walk through the "slow path" of calling max_pool1d_with_indices. Test Plan: - wait for tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/84127 Approved by: https://github.com/kshitij12345, https://github.com/samdow, https://github.com/malfet
Author
Committer
Parents
Loading