提交 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, ...@@ -173,9 +173,6 @@ def use(device, force=False, default_to_move_computation_to_gpu = True,
try: try:
if device !='gpu': if device !='gpu':
gpu_init(device) 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: if move_shared_float32_to_gpu:
handle_shared_float32(True) handle_shared_float32(True)
......
...@@ -1918,7 +1918,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args) ...@@ -1918,7 +1918,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args)
} }
if(card_number_provided) { 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); err = cudaSetDevice(card_nb);
if(cudaSuccess != err) { if(cudaSuccess != err) {
...@@ -1928,6 +1928,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args) ...@@ -1928,6 +1928,7 @@ CudaNdarray_gpu_init(PyObject* _unused, PyObject* args)
cudaGetErrorString(cudaGetLastError())); cudaGetErrorString(cudaGetLastError()));
} }
} }
fprintf(stderr, "Using gpu device %d: %s\n", card_nb, deviceProp.name);
Py_INCREF(Py_None); Py_INCREF(Py_None);
return Py_None; return Py_None;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论