mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
da3dd9af12
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/12390 Introduce a no op optimizer for when we don't want updates to happen, but don't want to affect downstream processes. Reviewed By: mlappelbaum Differential Revision: D10209812 fbshipit-source-id: 2af4ebc0fb42e78ea851c3a9f4860f3d224037b6
6 lines
164 B
Text
6 lines
164 B
Text
#include "caffe2/core/context_gpu.h"
|
|
#include "caffe2/operators/data_couple.h"
|
|
|
|
namespace caffe2 {
|
|
REGISTER_CUDA_OPERATOR(DataCouple, DataCoupleOp<CUDAContext>);
|
|
}
|