Skip to content

GitLab

Commit c4c96460 authored by Jing Zhang's avatar Jing Zhang
Browse files

format

parent f59394c2
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
example/01_gemm/gemm_xdl_input_f8_comp_f16.cpp
View file @ c4c96460
...@@ -21,9 +21,9 @@ ...@@ -21,9 +21,9 @@
template <ck::index_t... Is> template <ck::index_t... Is>
using S = ck::Sequence<Is...>; using S = ck::Sequence<Is...>;
using F16 = ck::half_t; using F16 = ck::half_t;
using F8 = ck::f8_t; using F8 = ck::f8_t;
using F32 = float; using F32 = float;
using Row = ck::tensor_layout::gemm::RowMajor; using Row = ck::tensor_layout::gemm::RowMajor;
using Col = ck::tensor_layout::gemm::ColumnMajor; using Col = ck::tensor_layout::gemm::ColumnMajor;
...@@ -52,8 +52,7 @@ struct f8_to_fp16 ...@@ -52,8 +52,7 @@ struct f8_to_fp16
struct Relu struct Relu
{ {
__host__ __device__ void __host__ __device__ void operator()(ck::half_t& e, const ck::half_t& c) const
operator()(ck::half_t& e, const ck::half_t& c) const
{ {
e = c > 0 ? c : 0; e = c > 0 ? c : 0;
} }
......
script/cmake-ck-dev.sh
View file @ c4c96460
...@@ -15,4 +15,5 @@ cmake ...@@ -15,4 +15,5 @@ cmake
-D GPU_TARGETS="gfx908;gfx90a;gfx940" \ -D GPU_TARGETS="gfx908;gfx90a;gfx940" \
-D CMAKE_VERBOSE_MAKEFILE:BOOL=ON \ -D CMAKE_VERBOSE_MAKEFILE:BOOL=ON \
-D USE_BITINT_EXTENSION_INT4=OFF \ -D USE_BITINT_EXTENSION_INT4=OFF \
-D DL_KERNELS=ON \
${MY_PROJECT_SOURCE} ${MY_PROJECT_SOURCE}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment