diff options
| author | Gregory Chanan <gchanan@fb.com> | 2019-04-01 07:57:48 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-01 08:00:33 -0700 |
| commit | 4c74cf74897d0fb05fe0060af6825eb779bd6f5c (patch) | |
| tree | 45e3b4359c28da84ba9927ecb6b27b6119d8c2a6 /caffe2 | |
| parent | ddbfdc911d94760bb431fcdf2a0322432c1c8272 (diff) | |
| download | pytorch-4c74cf74897d0fb05fe0060af6825eb779bd6f5c.tar.gz pytorch-4c74cf74897d0fb05fe0060af6825eb779bd6f5c.tar.bz2 pytorch-4c74cf74897d0fb05fe0060af6825eb779bd6f5c.zip | |
Move ideep singleton registration to ATen from C2. (#18335)
Summary:
Since we are going to add ideep to ATen, and ATen is always compiled, it makes sense to have the registration in ATen rather than C2.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18335
Reviewed By: bddppq
Differential Revision: D14578652
Pulled By: gchanan
fbshipit-source-id: 4d77fcfc21a362b21d5291a127498aa722548873
Diffstat (limited to 'caffe2')
| -rw-r--r-- | caffe2/ideep/utils/ideep_register.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/ideep/utils/ideep_register.cc b/caffe2/ideep/utils/ideep_register.cc index a35223802d..1932adcff5 100644 --- a/caffe2/ideep/utils/ideep_register.cc +++ b/caffe2/ideep/utils/ideep_register.cc @@ -1,7 +1,7 @@ #include <caffe2/core/event_cpu.h> #include <caffe2/core/operator.h> #include <caffe2/proto/caffe2_pb.h> -#include <ideep_pin_singletons.hpp> +#include <ideep/tensor.hpp> #include "ideep_context.h" namespace at { |