Commit dfbad58d authored by Junjie Bai's avatar Junjie Bai Committed by Facebook Github Bot

codemod cuda_gpu_id to device_id (#12022)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12022

codemod -d . --extensions h,cc,cpp,cu,py,proto,pbtxt,pb.txt,config cuda_gpu_id device_id

codemod with 'Yes to all'

Reviewed By: orionr

Differential Revision: D9986213

fbshipit-source-id: f5614a5d26078817aee8caf79a494abfd6a95ff1
parent d38ade04
...@@ -343,7 +343,7 @@ def get_device_option_cpu(): ...@@ -343,7 +343,7 @@ def get_device_option_cpu():
def get_device_option_cuda(gpu_id=0): def get_device_option_cuda(gpu_id=0):
device_option = caffe2_pb2.DeviceOption() device_option = caffe2_pb2.DeviceOption()
device_option.device_type = caffe2_pb2.CUDA device_option.device_type = caffe2_pb2.CUDA
device_option.cuda_gpu_id = gpu_id device_option.device_id = gpu_id
return device_option 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