96f7cdf1601ba703112f67838808aad2795ed9f7,theano/gpuarray/opt.py,,,#,1744

Before Change

                pdb.set_trace()

// Register the local_assert_no_cpu_op:
assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op,
                                            name="assert_no_cpu_op")
// 49.2 is after device specialization & fusion optimizations for last transfers
optdb.register("gpua_assert_no_cpu_op", assert_no_cpu_op, 49.2,
               "assert_no_cpu_op")


def tensor_to_gpu(x, context_name):
    if isinstance(x.type, tensor.TensorType):

After Change

pool_db2 = LocalGroupDB(local_opt=theano.gof.opt.GraphToGPULocalOptGroup)
pool_db2.__name__ = "pool_db2"
lifter = op_lifter([pool.Pool])(local_gpu_pool)
pool_db.register("local_gpu_pool", lifter,
                 "gpuarray", "fast_compile", "fast_run",
                 position=1)
pool_db2.register("local_gpu_pool",
                  local_optimizer([pool.Pool])(local_gpu_pool),
                  "gpuarray", "fast_compile", "fast_run",
                  position=1)

Instances