提交 e06617bd authored 作者: abergeron's avatar abergeron

Merge pull request #2915 from carriepl/scan_tests_new_backend

Move gpu tests from T_Scan to ScanGpuTests
...@@ -54,6 +54,14 @@ if pygpu: ...@@ -54,6 +54,14 @@ if pygpu:
optdb.add_tags('gpuarray_opt', 'fast_run', 'fast_compile', 'inplace') optdb.add_tags('gpuarray_opt', 'fast_run', 'fast_compile', 'inplace')
elif config.gpuarray.init_device != '': elif config.gpuarray.init_device != '':
init_dev(config.gpuarray.init_device) init_dev(config.gpuarray.init_device)
from .basic_ops import (GpuAlloc, GpuContiguous, GpuEye, GpuFromHost,
GpuJoin, GpuReshape, GpuSplit, HostFromGpu)
from .basic_ops import host_from_gpu, gpu_from_host
from .elemwise import GpuElemwise
from .subtensor import (GpuSubtensor, GpuIncSubtensor,
GpuAdvancedIncSubtensor1)
except Exception: except Exception:
error("Could not initialize pygpu, support disabled", exc_info=True) error("Could not initialize pygpu, support disabled", exc_info=True)
else: else:
......
...@@ -742,7 +742,7 @@ optdb.register('gpua_scanOp_make_inplace', ...@@ -742,7 +742,7 @@ optdb.register('gpua_scanOp_make_inplace',
scan_opt.ScanInplaceOptimizer(typeConstructor=GpuArrayType, scan_opt.ScanInplaceOptimizer(typeConstructor=GpuArrayType,
gpua_flag=True), gpua_flag=True),
75, 75,
'gpua', 'gpuarray',
'fast_run', 'fast_run',
'inplace', 'inplace',
'scan') 'scan')
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论