diff options
author | Hector Yuen <hyz@fb.com> | 2018-11-04 21:45:40 -0800 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-11-04 21:47:45 -0800 |
commit | 274f3c09518abe61061977f2681b47d1db8a3347 (patch) | |
tree | 7f26a0de6147a5f750d96c590a175a5a76a575a2 /caffe2/proto | |
parent | 246d5282b39f9536249bb85127f90857c976fa8d (diff) | |
download | pytorch-274f3c09518abe61061977f2681b47d1db8a3347.tar.gz pytorch-274f3c09518abe61061977f2681b47d1db8a3347.tar.bz2 pytorch-274f3c09518abe61061977f2681b47d1db8a3347.zip |
add explicit fpga context (#13318)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/13318
add a context to describe fpga
this will remove the need of having opencl with fpga engine
the next step is to change the opencl implementation to explicitly use the fpga context
Reviewed By: soumith
Differential Revision: D12828795
fbshipit-source-id: 0700a83672d117d7aa3d941cd39c2ae627cb6e5f
Diffstat (limited to 'caffe2/proto')
-rw-r--r-- | caffe2/proto/caffe2.proto | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/caffe2/proto/caffe2.proto b/caffe2/proto/caffe2.proto index a8acd9673e..e96ac2c7ef 100644 --- a/caffe2/proto/caffe2.proto +++ b/caffe2/proto/caffe2.proto @@ -166,7 +166,7 @@ message Argument { // DeviceType that Caffe2 currently supports. // Note: if you add a device type, make sure you add the corresponding device // line in the DeviceTypeName() function in caffe2/utils/proto_utils.cc -// and update ATen/core/DeviceType.h +// and update c10/DeviceType.h enum DeviceTypeProto { PROTO_CPU = 0; // In default, we will use CPU. PROTO_CUDA = 1; // CUDA. @@ -175,8 +175,9 @@ enum DeviceTypeProto { PROTO_OPENCL = 4; // OpenCL PROTO_IDEEP = 5; // IDEEP. PROTO_HIP = 6; // AMD HIP + PROTO_FPGA = 7; // FPGA // Change the following number if you add more devices in the code. - PROTO_COMPILE_TIME_MAX_DEVICE_TYPES = 7; + PROTO_COMPILE_TIME_MAX_DEVICE_TYPES = 8; PROTO_ONLY_FOR_TEST = 20901; // This device type is only for test. } |