| .. |
|
CMakeLists.txt
|
Append fp16 test case to verify Mma_HFMA2
|
2021-03-04 18:17:57 +08:00 |
|
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_dgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_simt_f16_sm50.cu
|
Append fp16 test case to verify Mma_HFMA2
|
2021-03-04 18:17:57 +08:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm50.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_fprop_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_problems.h
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_testbed_interleaved.h
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_testbed.h
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv2d_wgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_dgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_dgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_fprop_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_fprop_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_fprop_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_problems.h
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_testbed.h
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm75.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
|
conv3d_wgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
|
CUTLASS 2.5
|
2021-02-26 09:58:26 -05:00 |
