Example #1
0
 def __init__(self, name):
     T_subtensor.__init__(self, name,
                          shared=gpuarray_shared_constructor,
                          sub=GpuSubtensor,
                          mode=mode_with_gpu,
                          # avoid errors with limited devices
                          dtype='float32',
                          ignore_topo=(HostFromGpu,GpuFromHost,DeepCopyOp))
     assert self.sub == GpuSubtensor
Example #2
0
 def __init__(self, name):
     T_subtensor.__init__(self, name,
                          shared=gpuarray_shared_constructor,
                          sub=GpuSubtensor,
                          inc_sub=GpuIncSubtensor,
                          mode=mode_with_gpu,
                          # avoid errors with limited devices
                          dtype='float32',
                          ignore_topo=(HostFromGpu, GpuFromHost,
                                       DeepCopyOp))
     assert self.sub == GpuSubtensor
Example #3
0
 def __init__(self, name):
     T_subtensor.__init__(self, name,
                          shared=gpuarray_shared_constructor,
                          sub=GpuSubtensor,
                          inc_sub=GpuIncSubtensor,
                          mode=mode_with_gpu,
                          # avoid errors with limited devices
                          dtype='float32',
                          ignore_topo=(HostFromGpu, GpuFromHost,
                                       DeepCopyOp))
     # GPU opt can't run in fast_compile only.
     self.fast_compile = False
     assert self.sub == GpuSubtensor
Example #4
0
 def __init__(self, name):
     T_subtensor.__init__(
         self,
         name,
         shared=gpuarray_shared_constructor,
         sub=GpuSubtensor,
         inc_sub=GpuIncSubtensor,
         adv_incsub1=GpuAdvancedIncSubtensor1,
         mode=mode_with_gpu,
         # avoid errors with limited devices
         dtype='float32',
         ignore_topo=(HostFromGpu, GpuFromHost, DeepCopyOp))
     # GPU opt can't run in fast_compile only.
     self.fast_compile = False
     assert self.sub == GpuSubtensor