소스 검색

Make sure to take control of the right GPU

Samuel Thibault 13 년 전
부모
커밋
dfec7f881c
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      doc/chapters/advanced-examples.texi

+ 1 - 0
doc/chapters/advanced-examples.texi

@@ -666,6 +666,7 @@ be given the CUDA pointer at registration, for instance:
 
 	starpu_data_unregister(handle);
 
+	cudaSetDevice(starpu_worker_get_devid(workerid));
 	cudaGraphicsUnmapResources(1, &resource, 0);
 
 	/* Now display it */