You need to sign in or sign up before continuing.
Commit fb720bb0 authored by yuguo-Jack's avatar yuguo-Jack
Browse files

pybind.cc

parent 2304e2f0
......@@ -2372,7 +2372,7 @@ All parameter, weight, gradient are variables in Paddle.
},
py::return_value_policy::copy);
py::class_<gpuDeviceProp>(m, "_gpuDeviceProperties")
py::class_<gpuDeviceProp>(m, "_gpuDeviceProperties", py::module_local())
.def_property_readonly(
"name", [](const gpuDeviceProp &prop) { return prop.name; })
.def_property_readonly(
......
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