youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
cd39c75e250ef1c3dd11cd59dca4e53efe7d5ea8
cutlass/test/unit/conv/device
History
Andrew Kerr 8a951b2940 Enable convolution with fused epilogue for Volta Tensor Cores (#402) 
* Enabled convolution with epilogue fusion for Volta Tensor Cores.

* Compilation fixes

* Disabled testing Volta on Ampere architectures.
2022-01-30 23:24:50 -05:00
..
cache_testbed_output.h
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
CMakeLists.txt
Enable convolution with fused epilogue for Volta Tensor Cores (#402)
2022-01-30 23:24:50 -05:00
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
Updates to fused epilogue (#383)
2021-12-17 16:04:43 -05:00
conv2d_dgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_dgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_dgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_simt_f16_sm60.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
Enable convolution with fused epilogue for Volta Tensor Cores (#402)
2022-01-30 23:24:50 -05:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_fprop_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm50.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_qf32nhwc_qf32nhwc_qf32nhwc_simt_f32_sm50.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s4ncxhwx_s4cxrskx_s4ncxhwx_tensor_op_s32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s4nhwc_s4nhwc_s32nhwc_tensor_op_s32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s8ncxhwx_s8cxrskx_s8ncxhwx_tensor_op_s32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_s8nhwc_s8nhwc_s32nhwc_tensor_op_s32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_fprop_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_fprop_with_broadcast_sm70.cu
Enable convolution with fused epilogue for Volta Tensor Cores (#402)
2022-01-30 23:24:50 -05:00
conv2d_fprop_with_broadcast_sm75.cu
Add epilogue functor for residual block fusion (#391)
2021-12-29 22:53:40 -05:00
conv2d_fprop_with_reduction_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_problems.h
Updates to fused epilogue (#383)
2021-12-17 16:04:43 -05:00
conv2d_strided_dgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv2d_testbed_interleaved.h
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv2d_testbed.h
Updates to fused epilogue (#383)
2021-12-17 16:04:43 -05:00
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm50.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_wgrad_implicit_gemm_cf32nhwc_cf32nhwc_cf32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f16nhwc_tensor_op_f16_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm70.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm75.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_wgrad_implicit_gemm_f16nhwc_f16nhwc_f32nhwc_tensor_op_f32_sm80.cu
refine the implementation
2021-09-08 13:14:08 +00:00
conv2d_wgrad_implicit_gemm_f32nhwc_f32nhwc_f32nhwc_simt_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_wgrad_implicit_gemm_tf32nhwc_tf32nhwc_f32nhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv2d_with_broadcast_testbed.h
Add epilogue functor for residual block fusion (#391)
2021-12-29 22:53:40 -05:00
conv2d_with_reduction_testbed.h
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv3d_dgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_dgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv3d_fprop_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_fprop_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_fprop_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_problems.h
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_testbed.h
CUTLASS 2.7 (#318)
2021-09-20 11:02:22 -07:00
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm75.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_wgrad_implicit_gemm_f16ndhwc_f16ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00
conv3d_wgrad_implicit_gemm_tf32ndhwc_tf32ndhwc_f32ndhwc_tensor_op_f32_sm80.cu
Cutlass 2.6 Update 1 (#301)
2021-07-27 17:58:30 -07:00