Commit f7d394ce authored by Rick Ratmansky's avatar Rick Ratmansky Committed by Facebook Github Bot

Revert D10123245: Back out "codemod cuda_gpu_id to device_id"

Differential Revision:
D10123245

Original commit changeset: d83da8e00a12

fbshipit-source-id: fca91fea58b7df208edc2e218a1d514f9821ec7b
parent cbb0236d
......@@ -343,7 +343,7 @@ def get_device_option_cpu():
def get_device_option_cuda(gpu_id=0):
device_option = caffe2_pb2.DeviceOption()
device_option.device_type = caffe2_pb2.CUDA
device_option.cuda_gpu_id = gpu_id
device_option.device_id = gpu_id
return device_option
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment