Shaojie WANG 0d08cf1893 add GetWorkSpaceSize to base arg (#253) ...

* add GetWorkSpaceSize to base arg and make an example on convnd_bwd_weight

* remove redundant compute

* use datatype and split k to check whether a workspace is used

* remove unused computation for work space size

2022-05-24 11:13:00 -05:00

..

01_gemm

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

02_gemm_alpha_beta

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

03_gemm_bias_relu

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

04_gemm_bias_relu_add

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

06_conv2d_fwd_bias_relu

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

07_conv2d_fwd_bias_relu_add

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

09_convnd_fwd

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

10_conv2d_bwd_data

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

11_conv2d_bwd_weight

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

12_reduce

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

13_pool2d_fwd

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

14_gemm_xdl_requant_relu_requant

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

15_grouped_gemm

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

16_gemm_reduce

Gemm reduce max (#209)

2022-05-19 21:56:56 -05:00

17_convnd_bwd_data_xdl

Validate examples in CI (#233)

2022-05-13 16:54:44 -05:00

18_batched_gemm_reduce

Gemm reduce max (#209)

2022-05-19 21:56:56 -05:00

19_binary_elementwise

Hotfix eltiwseop (#242)

2022-05-19 22:02:06 -05:00

20_convnd_bwd_weight_xdl

add GetWorkSpaceSize to base arg (#253)

2022-05-24 11:13:00 -05:00

CMakeLists.txt

example of conv bwd weight 1d/2d/3d fp32/fp16/bf16 xdl (#244)

2022-05-20 17:20:10 -05:00