提交 6c365bf6 authored 作者: James Bergstra's avatar James Bergstra

cuda init: moved printing of active device to .cu file

上级 f5aa88a2
......@@ -173,9 +173,6 @@ def use(device, force=False, default_to_move_computation_to_gpu = True,
try:
if device !='gpu':
gpu_init(device)
else:
#warning To let people see that the gpu will be used.
_logger.warn("We let the driver select the gpu device to use")
if move_shared_float32_to_gpu:
handle_shared_float32(True)
......
......@@ -1918,7 +1918,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args)
}
if(card_number_provided) {
fprintf(stderr, "Using gpu device %d: %s\n", card_nb, deviceProp.name);
fprintf(stderr, "DEBUG: calling cudaSetDevice %i\n", card_nb);
err = cudaSetDevice(card_nb);
if(cudaSuccess != err) {
......@@ -1928,6 +1928,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args)
cudaGetErrorString(cudaGetLastError()));
}
}
fprintf(stderr, "Using gpu device %d: %s\n", card_nb, deviceProp.name);
Py_INCREF(Py_None);
return Py_None;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论