mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
68613cf5a2
Summary: This is an experimental build on top of what orionr and mingzhe09088 built. Essentially, the idea is that we will need separate *_API versions for different shared libraries. If this theory is right, I'll try to clean up the design a bit and document it properly. Pull Request resolved: https://github.com/pytorch/pytorch/pull/11266 Reviewed By: orionr Differential Revision: D9682942 Pulled By: Yangqing fbshipit-source-id: c79653199e67a1500c9174f39f8b0357324763f3
9 lines
226 B
Text
9 lines
226 B
Text
#include "caffe2/core/context_gpu.h"
|
|
#include "caffe2/operators/operator_fallback_gpu.h"
|
|
#include "caffe2/operators/sparse_normalize_op.h"
|
|
|
|
namespace caffe2 {
|
|
REGISTER_CUDA_OPERATOR(
|
|
SparseNormalize,
|
|
GPUFallbackOp);
|
|
}
|