| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| AbsKernel.cu | 2024-12-02 00:41 | 1.5K | ||
| Activation.cpp | 2024-12-02 00:41 | 3.4K | ||
| Activation.h | 2024-12-02 00:41 | 548 | ||
| ActivationEluKernel.cu | 2024-12-02 00:41 | 2.5K | ||
| ActivationGeluKernel.cu | 2024-12-02 00:41 | 3.8K | ||
| ActivationGluKernel.cu | 2024-12-02 00:41 | 4.5K | ||
| ActivationHardshrinkKernel.cu | 2024-12-02 00:41 | 1.0K | ||
| ActivationHardsigmoidKernel.cu | 2024-12-02 00:41 | 2.2K | ||
| ActivationHardswishKernel.cu | 2024-12-02 00:41 | 2.0K | ||
| ActivationHardtanhKernel.cu | 2024-12-02 00:41 | 1.3K | ||
| ActivationLeakyReluKernel.cu | 2024-12-02 00:41 | 1.8K | ||
| ActivationLogSigmoidKernel.cu | 2024-12-02 00:41 | 2.0K | ||
| ActivationMishKernel.cu | 2024-12-02 00:41 | 1.9K | ||
| ActivationPreluKernel.cu | 2024-12-02 00:41 | 1.4K | ||
| ActivationSiluKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| ActivationSoftplusKernel.cu | 2024-12-02 00:41 | 2.1K | ||
| ActivationSoftshrinkKernel.cu | 2024-12-02 00:41 | 1.6K | ||
| ActivationThresholdKernel.cu | 2024-12-02 00:41 | 1.2K | ||
| AdaptiveAveragePooling.cu | 2024-12-02 00:41 | 32K | ||
| AdaptiveAveragePooling3d.cu | 2024-12-02 00:41 | 17K | ||
| AdaptiveMaxPooling2d.cu | 2024-12-02 00:41 | 14K | ||
| AdaptiveMaxPooling3d.cu | 2024-12-02 00:41 | 15K | ||
| AmpKernels.cu | 2024-12-02 00:41 | 12K | ||
| AveragePool2d.cu | 2024-12-02 00:41 | 17K | ||
| AveragePool3d.cu | 2024-12-02 00:41 | 21K | ||
| BinaryBitwiseOpsKernels.cu | 2024-12-02 00:41 | 2.2K | ||
| BinaryDivFloorKernel.cu | 2024-12-02 00:41 | 2.9K | ||
| BinaryDivTrueKernel.cu | 2024-12-02 00:41 | 2.2K | ||
| BinaryDivTruncKernel.cu | 2024-12-02 00:41 | 1.8K | ||
| BinaryGeometricKernels.cu | 2024-12-02 00:41 | 1.2K | ||
| BinaryInternal.h | 2024-12-02 00:41 | 1.2K | ||
| BinaryLogicalOpsKernels.cu | 2024-12-02 00:41 | 4.2K | ||
| BinaryMiscBackwardOpsKernels.cu | 2024-12-02 00:41 | 4.8K | ||
| BinaryMiscOpsKernels.cu | 2024-12-02 00:41 | 2.8K | ||
| BinaryMulKernel.cu | 2024-12-02 00:41 | 1.6K | ||
| BinaryRemainderKernel.cu | 2024-12-02 00:41 | 2.0K | ||
| BinaryShiftOpsKernels.cu | 2024-12-02 00:41 | 1.5K | ||
| Blas.cpp | 2024-12-02 00:41 | 46K | ||
| Bucketization.cu | 2024-12-02 00:41 | 9.1K | ||
| CUDAJitLoops.cuh | 2024-12-02 00:41 | 11K | ||
| CUDALoops.cuh | 2024-12-02 00:41 | 11K | ||
| CUDAScalar.cu | 2024-12-02 00:41 | 2.1K | ||
| Col2Im.cu | 2024-12-02 00:41 | 4.5K | ||
| CompareEQKernel.cu | 2024-12-02 00:41 | 1.4K | ||
| CompareKernels.cu | 2024-12-02 00:41 | 3.0K | ||
| ComplexKernel.cu | 2024-12-02 00:41 | 1.1K | ||
| CompositeRandomAccessor.h | 2024-12-02 00:41 | 929 | ||
| ConvolutionMM2d.cu | 2024-12-02 00:41 | 16K | ||
| Copy.cu | 2024-12-02 00:41 | 14K | ||
| Copy.h | 2024-12-02 00:41 | 162 | ||
| CopysignKernel.cu | 2024-12-02 00:41 | 964 | ||
| CrossKernel.cu | 2024-12-02 00:41 | 3.2K | ||
| CuFFTPlanCache.h | 2024-12-02 00:41 | 18K | ||
| CuFFTUtils.h | 2024-12-02 00:41 | 1.8K | ||
| CumminmaxKernel.cu | 2024-12-02 00:41 | 1.3K | ||
| CumprodKernel.cu | 2024-12-02 00:41 | 643 | ||
| CumsumKernel.cu | 2024-12-02 00:41 | 647 | ||
| DepthwiseConv2d.cu | 2024-12-02 00:41 | 31K | ||
| DepthwiseConv3d.cu | 2024-12-02 00:41 | 28K | ||
| DeviceSqrt.cuh | 2024-12-02 00:41 | 573 | ||
| DilatedMaxPool2d.cu | 2024-12-02 00:41 | 24K | ||
| DilatedMaxPool3d.cu | 2024-12-02 00:41 | 22K | ||
| DistanceKernel.cu | 2024-12-02 00:41 | 16K | ||
| DistributionBernoulli.cu | 2024-12-02 00:41 | 1.4K | ||
| DistributionCauchyKernel.cu | 2024-12-02 00:41 | 558 | ||
| DistributionExponentialKernel.cu | 2024-12-02 00:41 | 557 | ||
| DistributionGeometricKernel.cu | 2024-12-02 00:41 | 541 | ||
| DistributionLogNormalKernel.cu | 2024-12-02 00:41 | 566 | ||
| DistributionNormal.cu | 2024-12-02 00:41 | 548 | ||
| DistributionRandomKernel.cu | 2024-12-02 00:41 | 1.2K | ||
| DistributionTemplates.h | 2024-12-02 00:41 | 27K | ||
| DistributionUniform.cu | 2024-12-02 00:41 | 540 | ||
| Distributions.cpp | 2024-12-02 00:41 | 2.8K | ||
| Distributions.cu | 2024-12-02 00:41 | 8.1K | ||
| Distributions.h | 2024-12-02 00:41 | 641 | ||
| Dropout.cu | 2024-12-02 00:41 | 15K | ||
| Embedding.cu | 2024-12-02 00:41 | 14K | ||
| EmbeddingBackwardKernel.cu | 2024-12-02 00:41 | 15K | ||
| EmbeddingBackwardKernel.cuh | 2024-12-02 00:41 | 543 | ||
| EmbeddingBag.cu | 2024-12-02 00:41 | 23K | ||
| Equal.cpp | 2024-12-02 00:41 | 1.6K | ||
| FillKernel.cu | 2024-12-02 00:41 | 920 | ||
| FlattenIndicesKernel.cu | 2024-12-02 00:41 | 745 | ||
| ForeachBinaryOpList.cu | 2024-12-02 00:41 | 17K | ||
| ForeachBinaryOpScalar.cu | 2024-12-02 00:41 | 9.3K | ||
| ForeachBinaryOpScalarList.cu | 2024-12-02 00:41 | 7.8K | ||
| ForeachBinaryOpScalarTensor.cu | 2024-12-02 00:41 | 8.0K | ||
| ForeachFunctors.cuh | 2024-12-02 00:41 | 22K | ||
| ForeachMinMaxFunctors.cuh | 2024-12-02 00:41 | 426 | ||
| ForeachPointwiseOp.cu | 2024-12-02 00:41 | 12K | ||
| ForeachReduceOp.cu | 2024-12-02 00:41 | 20K | ||
| ForeachTernaryOp.cu | 2024-12-02 00:41 | 4.9K | ||
| ForeachUnaryOp.cu | 2024-12-02 00:41 | 13K | ||
| FractionalMaxPool2d.cu | 2024-12-02 00:41 | 8.2K | ||
| FractionalMaxPool3d.cu | 2024-12-02 00:41 | 11K | ||
| FunctionOfAMatrixUtilsKernel.cu | 2024-12-02 00:41 | 3.3K | ||
| FusedAdamKernel.cu | 2024-12-02 00:41 | 4.6K | ||
| FusedAdamWKernel.cu | 2024-12-02 00:41 | 4.7K | ||
| FusedSgdKernel.cu | 2024-12-02 00:41 | 12K | ||
| GcdLcmKernel.cu | 2024-12-02 00:41 | 1.9K | ||
| GridSampler.cpp | 2024-12-02 00:41 | 3.1K | ||
| GridSampler.cu | 2024-12-02 00:41 | 44K | ||
| GridSampler.cuh | 2024-12-02 00:41 | 11K | ||
| GridSampler.h | 2024-12-02 00:41 | 1.1K | ||
| IGammaKernel.cu | 2024-12-02 00:41 | 29K | ||
| Im2Col.cu | 2024-12-02 00:41 | 4.3K | ||
| IndexKernel.cpp | 2024-12-02 00:41 | 2.9K | ||
| IndexKernel.cu | 2024-12-02 00:41 | 18K | ||
| IndexKernel.h | 2024-12-02 00:41 | 350 | ||
| Indexing.cu | 2024-12-02 00:41 | 75K | ||
| JitLoops.cuh | 2024-12-02 00:41 | 6.7K | ||
| KernelUtils.cuh | 2024-12-02 00:41 | 4.7K | ||
| LaunchUtils.h | 2024-12-02 00:41 | 306 | ||
| LegacyThrustHelpers.cu | 2024-12-02 00:41 | 4.3K | ||
| Lerp.cu | 2024-12-02 00:41 | 4.2K | ||
| LinearAlgebra.cu | 2024-12-02 00:41 | 4.4K | ||
| LinearAlgebraStubs.cpp | 2024-12-02 00:41 | 6.7K | ||
| LogAddExpKernel.cu | 2024-12-02 00:41 | 1.9K | ||
| LogcumsumexpKernel.cu | 2024-12-02 00:41 | 5.0K | ||
| Loops.cuh | 2024-12-02 00:41 | 11K | ||
| Loss.cu | 2024-12-02 00:41 | 22K | ||
| LossCTC.cu | 2024-12-02 00:41 | 38K | ||
| Math.cuh | 2024-12-02 00:41 | 120K | ||
| MaxMinElementwiseKernel.cu | 2024-12-02 00:41 | 3.3K | ||
| MaxUnpooling.cu | 2024-12-02 00:41 | 19K | ||
| MemoryAccess.cuh | 2024-12-02 00:41 | 13K | ||
| MiscUtils.h | 2024-12-02 00:41 | 958 | ||
| MixedDtypesLinear.cu | 2024-12-02 00:41 | 14K | ||
| MultiLabelMarginCriterion.cu | 2024-12-02 00:41 | 14K | ||
| MultiMarginLoss.cu | 2024-12-02 00:41 | 14K | ||
| MultiTensorApply.cuh | 2024-12-02 00:41 | 14K | ||
| MultinomialKernel.cu | 2024-12-02 00:41 | 15K | ||
| NLLLoss2d.cu | 2024-12-02 00:41 | 17K | ||
| NaiveConvolutionTranspose2d.cu | 2024-12-02 00:41 | 24K | ||
| NaiveConvolutionTranspose3d.cu | 2024-12-02 00:41 | 30K | ||
| NaiveDilatedConvolution.cu | 2024-12-02 00:41 | 19K | ||
| Nonzero.cu | 2024-12-02 00:41 | 5.7K | ||
| Normalization.cu | 2024-12-02 00:41 | 37K | ||
| Normalization.cuh | 2024-12-02 00:41 | 73K | ||
| PersistentSoftmax.cuh | 2024-12-02 00:41 | 18K | ||
| PointwiseOpsKernel.cu | 2024-12-02 00:41 | 6.3K | ||
| Pow.cuh | 2024-12-02 00:41 | 2.1K | ||
| PowKernel.cu | 2024-12-02 00:41 | 7.0K | ||
| RNN.cu | 2024-12-02 00:41 | 28K | ||
| Randperm.cu | 2024-12-02 00:41 | 5.2K | ||
| Randperm.cuh | 2024-12-02 00:41 | 2.1K | ||
| RangeFactories.cu | 2024-12-02 00:41 | 10K | ||
| RecordStream.cu | 2024-12-02 00:41 | 561 | ||
| Reduce.cu | 2024-12-02 00:41 | 1.7K | ||
| Reduce.cuh | 2024-12-02 00:41 | 47K | ||
| ReduceAMinMaxKernel.cu | 2024-12-02 00:41 | 1.6K | ||
| ReduceArgMaxKernel.cu | 2024-12-02 00:41 | 1.4K | ||
| ReduceArgMinKernel.cu | 2024-12-02 00:41 | 1.4K | ||
| ReduceLogicKernel.cu | 2024-12-02 00:41 | 1.2K | ||
| ReduceMaxValuesKernel.cu | 2024-12-02 00:41 | 1.8K | ||
| ReduceMinValuesKernel.cu | 2024-12-02 00:41 | 1.8K | ||
| ReduceMomentKernel.cu | 2024-12-02 00:41 | 2.9K | ||
| ReduceNormKernel.cu | 2024-12-02 00:41 | 2.4K | ||
| ReduceOps.cpp | 2024-12-02 00:41 | 3.3K | ||
| ReduceOps.h | 2024-12-02 00:41 | 500 | ||
| ReduceSumProdKernel.cu | 2024-12-02 00:41 | 6.8K | ||
| ReflectionPad.cu | 2024-12-02 00:41 | 22K | ||
| RenormKernel.cu | 2024-12-02 00:41 | 854 | ||
| Repeat.cu | 2024-12-02 00:41 | 1.9K | ||
| ReplicationPadding.cu | 2024-12-02 00:41 | 25K | ||
| Resize.cpp | 2024-12-02 00:41 | 2.4K | ||
| Resize.h | 2024-12-02 00:41 | 1.5K | ||
| RowwiseScaledMM.cu | 2024-12-02 00:41 | 18K | ||
| RowwiseScaledMM.h | 2024-12-02 00:41 | 361 | ||
| RreluWithNoise.cu | 2024-12-02 00:41 | 5.8K | ||
| ScanKernels.cpp | 2024-12-02 00:41 | 3.6K | ||
| ScanKernels.h | 2024-12-02 00:41 | 779 | ||
| ScanUtils.cuh | 2024-12-02 00:41 | 20K | ||
| ScatterGatherKernel.cu | 2024-12-02 00:41 | 19K | ||
| SegmentReduce.cu | 2024-12-02 00:41 | 22K | ||
| Shape.cu | 2024-12-02 00:41 | 19K | ||
| SoftMax.cu | 2024-12-02 00:41 | 49K | ||
| Sort.cpp | 2024-12-02 00:41 | 4.2K | ||
| Sort.cu | 2024-12-02 00:41 | 12K | ||
| Sort.h | 2024-12-02 00:41 | 409 | ||
| SortImpl.cu | 2024-12-02 00:41 | 1.2K | ||
| SortStable.cu | 2024-12-02 00:41 | 8.5K | ||
| SortStable.h | 2024-12-02 00:41 | 464 | ||
| SortUtils.cuh | 2024-12-02 00:41 | 12K | ||
| Sorting.cpp | 2024-12-02 00:41 | 6.2K | ||
| Sorting.cu | 2024-12-02 00:41 | 9.3K | ||
| Sorting.h | 2024-12-02 00:41 | 408 | ||
| SortingCommon.cuh | 2024-12-02 00:41 | 5.2K | ||
| SortingRadixSelect.cuh | 2024-12-02 00:41 | 12K | ||
| SparseBinaryOpIntersectionKernel.cu | 2024-12-02 00:41 | 7.2K | ||
| SparseMM.cu | 2024-12-02 00:41 | 770 | ||
| SpectralOps.cpp | 2024-12-02 00:41 | 19K | ||
| SpectralOps.cu | 2024-12-02 00:41 | 4.5K | ||
| StepKernel.cu | 2024-12-02 00:41 | 1.1K | ||
| SummaryOps.cu | 2024-12-02 00:41 | 15K | ||
| TensorCompare.cpp | 2024-12-02 00:41 | 816 | ||
| TensorCompare.cu | 2024-12-02 00:41 | 5.2K | ||
| TensorFactories.cu | 2024-12-02 00:41 | 15K | ||
| TensorModeKernel.cpp | 2024-12-02 00:41 | 3.4K | ||
| TensorModeKernel.cu | 2024-12-02 00:41 | 9.6K | ||
| TensorModeKernel.cuh | 2024-12-02 00:41 | 14K | ||
| TensorModeKernel.h | 2024-12-02 00:41 | 431 | ||
| TensorShape.cu | 2024-12-02 00:41 | 30K | ||
| TensorShapeCUDA.cpp | 2024-12-02 00:41 | 1.4K | ||
| TensorTopK.cpp | 2024-12-02 00:41 | 3.3K | ||
| TensorTopK.cu | 2024-12-02 00:41 | 37K | ||
| TensorTopK.h | 2024-12-02 00:41 | 266 | ||
| TensorTransformations.cu | 2024-12-02 00:41 | 4.9K | ||
| TriangularOps.cu | 2024-12-02 00:41 | 6.0K | ||
| UnaryComplexKernels.cu | 2024-12-02 00:41 | 3.0K | ||
| UnaryFractionKernels.cu | 2024-12-02 00:41 | 6.7K | ||
| UnaryGammaKernels.cu | 2024-12-02 00:41 | 4.3K | ||
| UnaryGeometricAcosKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricAcoshKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricAsinKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricAsinhKernel.cu | 2024-12-02 00:41 | 1.8K | ||
| UnaryGeometricAtanKernel.cu | 2024-12-02 00:41 | 1.6K | ||
| UnaryGeometricAtanhKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricCosKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricCoshKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricSinKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricSinhKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricTanKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryGeometricTanhKernel.cu | 2024-12-02 00:41 | 1.7K | ||
| UnaryLogKernels.cu | 2024-12-02 00:41 | 4.1K | ||
| UnaryOpsKernel.cu | 2024-12-02 00:41 | 9.9K | ||
| UnarySignKernels.cu | 2024-12-02 00:41 | 4.4K | ||
| UnarySpecialOpsKernel.cu | 2024-12-02 00:41 | 15K | ||
| UnfoldBackwardKernel.cu | 2024-12-02 00:41 | 5.0K | ||
| Unique.cu | 2024-12-02 00:41 | 8.6K | ||
| UniqueCub.cu | 2024-12-02 00:41 | 11K | ||
| UniqueCub.cuh | 2024-12-02 00:41 | 352 | ||
| UpSample.cuh | 2024-12-02 00:41 | 11K | ||
| UpSampleBicubic2d.cu | 2024-12-02 00:41 | 9.8K | ||
| UpSampleBilinear2d.cu | 2024-12-02 00:41 | 35K | ||
| UpSampleLinear1d.cu | 2024-12-02 00:41 | 7.4K | ||
| UpSampleNearest1d.cu | 2024-12-02 00:41 | 8.4K | ||
| UpSampleNearest2d.cu | 2024-12-02 00:41 | 18K | ||
| UpSampleNearest3d.cu | 2024-12-02 00:41 | 12K | ||
| UpSampleTrilinear3d.cu | 2024-12-02 00:41 | 14K | ||
| ValidateCompressedIndicesKernel.cu | 2024-12-02 00:41 | 667 | ||
| WeightNorm.cu | 2024-12-02 00:41 | 17K | ||
| ZetaKernel.cu | 2024-12-02 00:41 | 1.3K | ||
| airy_ai.cu | 2024-12-02 00:41 | 1.2K | ||
| bessel_j0.cu | 2024-12-02 00:41 | 1.3K | ||
| bessel_j1.cu | 2024-12-02 00:41 | 1.3K | ||
| bessel_y0.cu | 2024-12-02 00:41 | 1.4K | ||
| bessel_y1.cu | 2024-12-02 00:41 | 1.4K | ||
| block_reduce.cuh | 2024-12-02 00:41 | 4.2K | ||
| chebyshev_polynomial_t.cu | 2024-12-02 00:41 | 1.3K | ||
| chebyshev_polynomial_u.cu | 2024-12-02 00:41 | 1.3K | ||
| chebyshev_polynomial_v.cu | 2024-12-02 00:41 | 1.3K | ||
| chebyshev_polynomial_w.cu | 2024-12-02 00:41 | 1.3K | ||
| cutlass_extensions/ | 2024-12-02 00:41 | - | ||
| fused_adam_amsgrad_impl.cu | 2024-12-02 00:41 | 3.1K | ||
| fused_adam_amsgrad_impl.cuh | 2024-12-02 00:41 | 1.0K | ||
| fused_adam_impl.cu | 2024-12-02 00:41 | 2.9K | ||
| fused_adam_impl.cuh | 2024-12-02 00:41 | 966 | ||
| fused_adam_utils.cuh | 2024-12-02 00:41 | 6.9K | ||
| fused_adamw_amsgrad_impl.cu | 2024-12-02 00:41 | 3.1K | ||
| fused_adamw_amsgrad_impl.cuh | 2024-12-02 00:41 | 1.0K | ||
| fused_adamw_impl.cu | 2024-12-02 00:41 | 3.0K | ||
| fused_adamw_impl.cuh | 2024-12-02 00:41 | 968 | ||
| group_norm_kernel.cu | 2024-12-02 00:41 | 33K | ||
| hermite_polynomial_h.cu | 2024-12-02 00:41 | 1.3K | ||
| hermite_polynomial_he.cu | 2024-12-02 00:41 | 1.3K | ||
| im2col.cuh | 2024-12-02 00:41 | 9.7K | ||
| int4mm.cu | 2024-12-02 00:41 | 46K | ||
| jit_utils.cpp | 2024-12-02 00:41 | 55K | ||
| jit_utils.h | 2024-12-02 00:41 | 6.0K | ||
| laguerre_polynomial_l.cu | 2024-12-02 00:41 | 1.3K | ||
| layer_norm_kernel.cu | 2024-12-02 00:41 | 50K | ||
| legendre_polynomial_p.cu | 2024-12-02 00:41 | 1.3K | ||
| linalg/ | 2024-12-02 00:41 | - | ||
| modified_bessel_i0.cu | 2024-12-02 00:41 | 1.5K | ||
| modified_bessel_i1.cu | 2024-12-02 00:41 | 1.5K | ||
| modified_bessel_k0.cu | 2024-12-02 00:41 | 1.5K | ||
| modified_bessel_k1.cu | 2024-12-02 00:41 | 1.5K | ||
| reduction_template.cuh | 2024-12-02 00:41 | 21K | ||
| scaled_modified_bessel_k0.cu | 2024-12-02 00:41 | 1.5K | ||
| scaled_modified_bessel_k1.cu | 2024-12-02 00:41 | 1.5K | ||
| shifted_chebyshev_polynomial_t.cu | 2024-12-02 00:41 | 1.4K | ||
| shifted_chebyshev_polynomial_u.cu | 2024-12-02 00:41 | 1.4K | ||
| shifted_chebyshev_polynomial_v.cu | 2024-12-02 00:41 | 1.3K | ||
| shifted_chebyshev_polynomial_w.cu | 2024-12-02 00:41 | 1.4K | ||
| spherical_bessel_j0.cu | 2024-12-02 00:41 | 1.5K | ||
| thread_constants.h | 2024-12-02 00:41 | 611 | ||
| vol2col.cuh | 2024-12-02 00:41 | 7.9K | ||