mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
52b8a52e4d
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/31281 Reviewed By: houseroad Differential Revision: D19053453 fbshipit-source-id: 350bfd5c001db9c17916dcae7ade8f56db1e9841
13 lines
301 B
Plaintext
13 lines
301 B
Plaintext
#include "caffe2/core/context_gpu.h"
|
|
#include "caffe2/operators/alias_with_name.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
REGISTER_CUDA_OPERATOR(AliasWithName, AliasWithNameOp<CUDAContext>);
|
|
|
|
} // namespace caffe2
|
|
|
|
C10_EXPORT_CAFFE2_OP_TO_C10_CUDA(
|
|
AliasWithName,
|
|
caffe2::AliasWithNameOp<caffe2::CUDAContext>);
|