You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

ashok 169dbcc38d Initial commit		4 weeks ago
..
Activation.h	Initial commit	4 weeks ago
BinaryInternal.h	Initial commit	4 weeks ago
CUDAJitLoops.cuh	Initial commit	4 weeks ago
CUDALoops.cuh	Initial commit	4 weeks ago
CompositeRandomAccessor.h	Initial commit	4 weeks ago
Copy.h	Initial commit	4 weeks ago
CuFFTPlanCache.h	Initial commit	4 weeks ago
CuFFTUtils.h	Initial commit	4 weeks ago
DeviceSqrt.cuh	Initial commit	4 weeks ago
DistributionTemplates.h	Initial commit	4 weeks ago
Distributions.h	Initial commit	4 weeks ago
EmbeddingBackwardKernel.cuh	Initial commit	4 weeks ago
ForeachFunctors.cuh	Initial commit	4 weeks ago
ForeachMinMaxFunctors.cuh	Initial commit	4 weeks ago
GridSampler.cuh	Initial commit	4 weeks ago
GridSampler.h	Initial commit	4 weeks ago
GroupMM.h	Initial commit	4 weeks ago
GroupMMCommon.cuh	Initial commit	4 weeks ago
IndexKernel.h	Initial commit	4 weeks ago
IndexKernelUtils.h	Initial commit	4 weeks ago
JitLoops.cuh	Initial commit	4 weeks ago
KernelUtils.cuh	Initial commit	4 weeks ago
LaunchUtils.h	Initial commit	4 weeks ago
Loops.cuh	Initial commit	4 weeks ago
Math.cuh	Initial commit	4 weeks ago
MemoryAccess.cuh	Initial commit	4 weeks ago
MiscUtils.h	Initial commit	4 weeks ago
MultiTensorApply.cuh	Initial commit	4 weeks ago
Normalization.cuh	Initial commit	4 weeks ago
PersistentSoftmax.cuh	Initial commit	4 weeks ago
Pow.cuh	Initial commit	4 weeks ago
Randperm.cuh	Initial commit	4 weeks ago
Reduce.cuh	Initial commit	4 weeks ago
ReduceOps.h	Initial commit	4 weeks ago
Resize.h	Initial commit	4 weeks ago
RowwiseScaledMM.h	Initial commit	4 weeks ago
ScaledGroupMM.h	Initial commit	4 weeks ago
ScanKernels.h	Initial commit	4 weeks ago
ScanUtils.cuh	Initial commit	4 weeks ago
Sort.h	Initial commit	4 weeks ago
SortStable.h	Initial commit	4 weeks ago
SortUtils.cuh	Initial commit	4 weeks ago
Sorting.h	Initial commit	4 weeks ago
SortingCommon.cuh	Initial commit	4 weeks ago
SortingRadixSelect.cuh	Initial commit	4 weeks ago
TensorModeKernel.cuh	Initial commit	4 weeks ago
TensorModeKernel.h	Initial commit	4 weeks ago
TensorTopK.h	Initial commit	4 weeks ago
UniqueCub.cuh	Initial commit	4 weeks ago
UpSample.cuh	Initial commit	4 weeks ago
block_reduce.cuh	Initial commit	4 weeks ago
cuBlasCommonArgs.h	Initial commit	4 weeks ago
cutlass_common.cuh	Initial commit	4 weeks ago
fused_adagrad_impl.cuh	Initial commit	4 weeks ago
fused_adagrad_utils.cuh	Initial commit	4 weeks ago
fused_adam_amsgrad_impl.cuh	Initial commit	4 weeks ago
fused_adam_impl.cuh	Initial commit	4 weeks ago
fused_adam_utils.cuh	Initial commit	4 weeks ago
fused_adamw_amsgrad_impl.cuh	Initial commit	4 weeks ago
fused_adamw_impl.cuh	Initial commit	4 weeks ago
im2col.cuh	Initial commit	4 weeks ago
jit_utils.h	Initial commit	4 weeks ago
reduction_template.cuh	Initial commit	4 weeks ago
thread_constants.h	Initial commit	4 weeks ago
vol2col.cuh	Initial commit	4 weeks ago