5dc3e70446c09312e2bd6dbe263b9af6f8405e62,scikits/cuda/special.py,,exp1,#,145

Before Change

    // registers to be invoked more threads per block:
    dev = misc.get_current_device()
    max_threads_per_block = 256
    block_dim, grid_dim = \
               misc.select_block_grid_sizes(dev, z_gpu.shape, max_threads_per_block)

    // Set this to False when debugging to make sure the compiled kernel is

After Change

        raise ValueError("unsupported type")
    op = "e[i] = exp1(z[i])"
    
    try:
        func = exp1.cache[z_gpu.dtype]
    except KeyError:
        func = elementwise.ElementwiseKernel(args, op,
                                 options=["-I", install_headers],
                                 preamble="//include "cuSpecialFuncs.h"")
        exp1.cache[z_gpu.dtype] = func

    e_gpu = gpuarray.empty_like(z_gpu)
    func(z_gpu, e_gpu)

    return e_gpu

Instances