提交 5a47c4b6 authored 作者: Frederic's avatar Frederic

Add a check when we init the gpu about the warpSize != 32.

I kept the check elsewhere to allow finding where this could cause problem more easily.
上级 330f40aa
......@@ -373,7 +373,12 @@ def use(device,
if test_driver:
import theano.sandbox.cuda.tests.test_driver
theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
if device_properties(use.device_number)["warpSize"] != 32:
raise ValueError("Your GPU have a warpSize of 32. Currently"
" we have code that depend on this. Email"
" Theano mailing list to tell us about"
" this new GPU as we don't know any with"
" this properties")
if move_shared_float32_to_gpu:
handle_shared_float32(True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论