mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00

Files

Po Yen Chen 4a2a56c22f Rangify constructor of HostTensorDescriptor & Tensor<> (#445 )

* Rangify STL algorithms

This commit adapts rangified std::copy(), std::fill() & std::transform()

* Rangify check_err()

By rangifying check_err(), we can not only compare values between
std::vector<>s, but also compare any ranges which have same value
type.

* Allow constructing Tensor<> like a HostTensorDescriptor

* Simplify Tensor<> object construction logics

* Remove more unnecessary 'HostTensorDescriptor' objects

* Re-format example code

* Re-write more HostTensorDescriptor ctor call

2022-11-11 11:36:01 -06:00

CMakeLists.txt

Standalone softmax kernel (#284 )

2022-06-21 14:59:19 -05:00

README.md

Standalone softmax kernel (#284 )

2022-06-21 14:59:19 -05:00

softmax_blockwise.cpp

Rangify constructor of HostTensorDescriptor & Tensor<> (#445 )

2022-11-11 11:36:01 -06:00

README.md

Instructions for `example_softmax_blockwise`

Run `example_softmax_blockwise`

# -D <xxx> : input 3-d tensor lengths
# -v <x> :   verification (0=no, 1=yes)
#arg1: initialization (0=no init, 1=single integer value, 2=scope integer value, 3=decimal value)
#arg2: time kernel (0=no, 1=yes)
example_softmax_blockwise -D 4,128,2048 -v 1 1 1

Result

launch_and_time_kernel: grid_dim {64, 1, 1}, block_dim {256, 1, 1}
Warm up 1 time
Start running 10 times...
Perf: 0.0242877 ms, 259.039 GB/s, DeviceReduceSoftmax<256,M_C8_S1,K_C32_S8,InSrcVectorDim_1_InSrcVectorSize_8_OutDstVectorSize_8>

README.md

Instructions for example_softmax_blockwise

Run example_softmax_blockwise

Instructions for `example_softmax_blockwise`

Run `example_softmax_blockwise`