pytorch
3bd250fd - [nnc] Test ability to vectorize reads from an intermediate tensor (#53752)

Commit
4 years ago
[nnc] Test ability to vectorize reads from an intermediate tensor (#53752) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53752 This test doesn't work today because we don't properly vectorize "FunctionCall" (which is the way one accesses an intermediate tensor). ghstack-source-id: 123592860 Test Plan: `buck test //caffe2/test/cpp/tensorexpr -- LoopNest.VectorizeUse` Reviewed By: ZolotukhinM Differential Revision: D26895550 fbshipit-source-id: 0798ebf3e6a834bd70181732c81528455d5329fa
Author
Parents
Loading