OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Make caffe2/fb folder compatible with AMD (#29131)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29131

caffe2_pb2.CUDA --> workspace.GpuDeviceType
workspace.NumCudaDevices() --> workspace.NumGpuDevices()

Also added the totalGlobalMem into get_device_properties(), which is needed by multi_gpu_utils.py

Test Plan:
sandcastle

f148921769

Reviewed By: bddppq

Differential Revision: D18290090

fbshipit-source-id: bde7c175d1fb6ff59a062266c1b17de39d113b24
This commit is contained in:
Xiaodong Wang 2019-11-04 16:38:32 -08:00 committed by Facebook Github Bot
parent 02e34919ae
commit cb72c9f5b1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
caffe2/python/pybind_state_hip.cc
View File

@ -47,6 +47,7 @@ void addHIPGlobalMethods(py::module& m) {
obj["name"] = py::cast(prop.name);
obj["major"] = py::cast(prop.major);
obj["minor"] = py::cast(prop.minor);
obj["totalGlobalMem"] = py::cast(prop.totalGlobalMem);
return obj;
});
};