diff options
| author | Bram Wasti <bwasti@fb.com> | 2018-04-20 11:31:21 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-04-20 11:31:21 -0700 |
| commit | a73b3fd1f0f4125fb2086b2c1af0ecf148a170b8 (patch) | |
| tree | b7f44f3d0ba7916975c979b4d4f5dc9300f74552 /caffe2/proto | |
| parent | 8a15bc4c9c3b7aa13e0df47e44539a3b3243a2c3 (diff) | |
| download | pytorch-a73b3fd1f0f4125fb2086b2c1af0ecf148a170b8.tar.gz pytorch-a73b3fd1f0f4125fb2086b2c1af0ecf148a170b8.tar.bz2 pytorch-a73b3fd1f0f4125fb2086b2c1af0ecf148a170b8.zip | |
[caffe2][opencl] Add OpenCL context (#6777)
Diffstat (limited to 'caffe2/proto')
| -rw-r--r-- | caffe2/proto/caffe2.proto | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/caffe2/proto/caffe2.proto b/caffe2/proto/caffe2.proto index 08df2b4f9d..dc3b0397f5 100644 --- a/caffe2/proto/caffe2.proto +++ b/caffe2/proto/caffe2.proto @@ -114,8 +114,9 @@ enum DeviceType { CUDA = 1; // CUDA. MKLDNN = 2; // Reserved for explicit MKLDNN OPENGL = 3; // OpenGL + OPENCL = 4; // OpenCL // Change the following number if you add more devices in the code. - COMPILE_TIME_MAX_DEVICE_TYPES = 4; + COMPILE_TIME_MAX_DEVICE_TYPES = 5; ONLY_FOR_TEST = 20901701; // This device type is only for test. } |