mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-20 21:09:08 +00:00
[Ck_tile] smoothquant (#1617)
* fix compile error
* fix typo of padding
* Add smoothquant op
* Add smoothquant instance library
* refine type
* add test script
* Re-generate smoothquant.hpp
* Always use 'current year' in copyright
* use Generic2dBlockShape instead
* Add vector = 8 instance back
* Find exe path automatically
* Simplify the api condition
* Remove debugging code
* update year
* Add blank line between function declaration
* explicitly cast return value to dim3
* refine return value
* Fix default warmup and repeat value
* Add comment
* refactor sommthquant cmake
* Add README
* Fix typo
---------
Co-authored-by: Po Yen, Chen <PoYen.Chen@amd.com>
[ROCm/composable_kernel commit: fbd654545a]
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
|
||||
# run from top of ck folder
|
||||
EXE=build/bin/tile_example_layernorm2d_fwd
|
||||
#!/bin/sh
|
||||
EXE="$(find . -name tile_example_layernorm2d_fwd -type f | head -n 1)"
|
||||
|
||||
$EXE -m=1 -n=1 -e=1e-12 -v=1 -prec_i=bf16 -repeat=1000
|
||||
$EXE -m=700 -n=80 -e=1e-12 -v=1 -prec_i=bf16 -repeat=1000
|
||||
|
||||
@@ -1,6 +1,5 @@
|
||||
#!/bin/sh
|
||||
# call from top of CK folder
|
||||
EXE=./build/bin/tile_example_layernorm2d_fwd
|
||||
EXE="$(find . -name tile_example_layernorm2d_fwd -type f | head -n 1)"
|
||||
|
||||
for fquant in "" "-fquant=1 -prec_o=int8"; do
|
||||
for pr_i in "fp16" "bf16" ; do
|
||||
|
||||
Reference in New Issue
Block a user