summaryrefslogtreecommitdiff
path: root/caffe2/proto
diff options
context:
space:
mode:
authorHector Yuen <hyz@fb.com>2018-11-04 21:45:40 -0800
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>2018-11-04 21:47:45 -0800
commit274f3c09518abe61061977f2681b47d1db8a3347 (patch)
tree7f26a0de6147a5f750d96c590a175a5a76a575a2 /caffe2/proto
parent246d5282b39f9536249bb85127f90857c976fa8d (diff)
downloadpytorch-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.proto5
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.
}