-
Notifications
You must be signed in to change notification settings - Fork 17
Description
The following installation for Composable Kernel Library does not even work because there is no cpu_avx2 branch.
git clone https://github.com/ROCmSoftwarePlatform/composable_kernel.git
cd composable_kernel
git checkout origin/cpu_avx2 -b cpu_avx2
Below are full text from this project's README on how to compile. It does't work.
Composable Kernel Library Installation
Composable Kernel aims to provide a programming model for writing performance critical kernels for machine learning workloads across multiple architectures including GPUs, CPUs, etc, through general purpose kernel languages, like HIP C++. Composable Kernel can be downloaded from the AMD ROCm Software Platform Repository (https://github.com/ROCmSoftwarePlatform/composable_kernel).
Composable Kernel Library Setup
Composable Kernel (CK) for CPU is currently only on the cpu_avx2 branch of the Composable Kernel repository and is at the experimental stage of development.
Prerequisites
CK is suitable for these compilers:
hipclang: this is mainly used for compiling GPU hip kernels(require rocm environment), but also can be used for CPU. For a first trial use below compiler.
gcc: at least gcc-12.2 is needed, you may need to manually install a gcc-12.2 if default is lower than gcc-12.2.
Download code
git clone https://github.com/ROCmSoftwarePlatform/composable_kernel.git
cd composable_kernel
git checkout origin/cpu_avx2 -b cpu_avx2
Compile
From the root directory of Composable Kernel (CK) build CK libraries:
if use gcc
sh script/cmake-avx2-gcc.sh
cd build
make -jnprocexample_cpu_conv2d_fwd example_cpu_conv2d_fwd_bias_relu_add