mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-26 08:00:13 +00:00

Files

Chao Liu 7a98e9fa34 N-D Tensor Contraction example, instance, and client example (#270 )

* adding contraction

* add contraction example

* update examle

* update example

* format

* update readme

* clean header

* clean header

* contraction with multiple D

* rename

* fix naming issue; add instances for contraction+bilinear

* change assumed virtual layout of contraction; add client example

* update example

* update

* contraction+scale

* use type_convert

* rename

[ROCm/composable_kernel commit: 4fe9c393b8]

2022-07-07 14:31:11 -05:00

CMakeLists.txt

Gemm+Bilinear (#316 )

2022-07-02 09:15:38 -05:00

gemm_bias_relu_xdl_fp16.cpp

N-D Tensor Contraction example, instance, and client example (#270 )

2022-07-07 14:31:11 -05:00

README.md

Gemm+Bilinear (#316 )

2022-07-02 09:15:38 -05:00

README.md

Instructions for `example_gemm_bias_relu_xdl_fp16`

Run `example_gemm_bias_relu_xdl_fp16`

#arg1: verification (0=no, 1=yes)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg3: time kernel (0=no, 1=yes)
#arg4 to 9: M (256x), N(128x), K(32x), StrideA, StrideB, StrideE
./bin/example_gemm_bias_relu_xdl_fp16 1 1 1 3840 4096 4096 4096 4096 4096

README.md

Instructions for example_gemm_bias_relu_xdl_fp16

Run example_gemm_bias_relu_xdl_fp16

Instructions for `example_gemm_bias_relu_xdl_fp16`

Run `example_gemm_bias_relu_xdl_fp16`