leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
ccb697bac77fcc898e9c897b2c90aa5b60ac72fb
cutlass/test/unit/conv/device
History
Manish Gupta 6615010cd0 CUTLASS 2.4 (Implicit GEMM convolution) (#147) 
CUTLASS 2.4 (Implicit GEMM Convolution)

Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-19 21:25:25 -08:00
..
CMakeLists.txt
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_dgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_fprop_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_problems.h
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_testbed_interleaved.h
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_testbed.h
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv2d_wgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_dgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_fprop_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_problems.h
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_testbed.h
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00
conv3d_wgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147)
2020-11-19 21:25:25 -08:00