• Arnaud Bergeron's avatar
    Rework the interface to GpuKernelBase to accept a list of kernel object. · a8b3b329
    Arnaud Bergeron 提交于
    Each item will be precompiled separetly and embedded into the c_code
    of the Op. This allows ops that need multiple kernels or that will
    choose between alternatives at runtime to use this interface.  It also
    groups all kernel-related parameters under one object.
    
    This change also save the source of the kernel code to re-attempt
    source compilation in case the binary is rejected for some reason
    (some implementations do not support reloading from pre-compiled
    kernel).
    
    There may still be more changes to how stuff works under the hood
    (most notably a blacklist of bad runtime/drivers that crash when
    attempting to create a kernel from a binary), but the visible
    interface should not change anymore, so now is the time to start using
    it more.
    a8b3b329
名称
最后提交
最后更新
..
tests 正在载入提交数据...
__init__.py 正在载入提交数据...
basic_ops.py 正在载入提交数据...
blas.py 正在载入提交数据...
conv.cu 正在载入提交数据...
conv.py 正在载入提交数据...
conv_full_kernel.cu 正在载入提交数据...
conv_kernel.cu 正在载入提交数据...
elemwise.py 正在载入提交数据...
kernel_codegen.py 正在载入提交数据...
neighbours.py 正在载入提交数据...
nnet.py 正在载入提交数据...
opt.py 正在载入提交数据...
subtensor.py 正在载入提交数据...
type.py 正在载入提交数据...