pytorch
ef78fa86 - caffe2::OperatorBase do not need to be aware of at::Tensor functions (#34810)

Commit
4 years ago
caffe2::OperatorBase do not need to be aware of at::Tensor functions (#34810) Summary: Replacing <ATen/core/Tensor.h> with <<ATen/core/TensorBody.h> speeds up compilation of caffe2 operators by 15% For example, it reduces pool_op.cu compilation from 18.8s to 16s Pull Request resolved: https://github.com/pytorch/pytorch/pull/34810 Test Plan: CI Differential Revision: D20472230 Pulled By: malfet fbshipit-source-id: e1b261cc24ff577f09e2d5f6428be2063c6d4a8b
Author
Parents
Loading