Skip to content

[CPU][RV64] Enhance RV64 convolution with GEMM optimizations and refactoring #312

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
strelkovkm wants to merge 16 commits into
base: v3.10_for_ie_master
Choose a base branch
from
Open

[CPU][RV64] Enhance RV64 convolution with GEMM optimizations and refactoring #312

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
461218e
[CPU] rv64: Add rvv_gemm_convolution.cpp
strelkovkm Mar 13, 2026
6e5fca8
[CPU][RV64] Add: xbyak_riscv, jit_rvv_1x1
strelkovkm Mar 20, 2026
231fcc0
[CPU][RV64] Edit:CMakeLists.txt
strelkovkm Mar 23, 2026
a51ec7d
[CPU][RV64] Edit:CMake src/
strelkovkm Apr 22, 2026
3f04423
[CPU][RV64] Edit: Return ref impl src/
strelkovkm Apr 22, 2026
981f682
[CPU][RV64] optimize jit_rvv & im2col
strelkovkm Apr 22, 2026
9b7837f
[CPU][RV64] bugfix
strelkovkm Apr 22, 2026
19f0502
Merge branch 'v3.10_for_ie_master' into dev_conv_rv64
strelkovkm Apr 24, 2026
e55a495
Update branch
strelkovkm Apr 24, 2026
874b0f3
Reset
strelkovkm Apr 24, 2026
9faf455
Merge branch 'openvinotoolkit:v3.10_for_ie_master' into v3.10_for_ie_…
strelkovkm Apr 24, 2026
a551c01
Merge branch 'v3.10_for_ie_master' into dev_conv_rv64
strelkovkm Apr 24, 2026
a9d51bf
[CPU][RV64]Return ref & add debug comments
strelkovkm Apr 26, 2026
7c82da9
[CPU][RV64] Optimize im2col, RVV(Bias + ReLU)
strelkovkm Apr 26, 2026
9b4e111
[CPU][RV64] Return ref, clean comments and edit choose impl
strelkovkm Apr 28, 2026
6791f0e
[CPU][RV64] Clang-formated
strelkovkm Apr 28, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/cpu/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ foreach(SOURCE_FILE ${SOURCES_EXTRA})
list(APPEND SOURCES "${SOURCE_FILE}")
endforeach()

if((DNNL_TARGET_ARCH STREQUAL "X64") OR (DNNL_TARGET_ARCH STREQUAL "AARCH64"))
if((DNNL_TARGET_ARCH STREQUAL "X64") OR (DNNL_TARGET_ARCH STREQUAL "AARCH64") OR (DNNL_TARGET_ARCH STREQUAL "RV64"))
file(GLOB_RECURSE SOURCES_JIT_UTILS
${CMAKE_CURRENT_SOURCE_DIR}/jit_utils/*.[ch]
${CMAKE_CURRENT_SOURCE_DIR}/jit_utils/*.[ch]pp
Expand Down
7 changes: 7 additions & 0 deletions src/cpu/cpu_convolution_list.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,9 @@ using namespace dnnl::impl::cpu::aarch64;
#include "cpu/acl/acl_indirect_gemm_convolution.hpp"
#include "cpu/acl/acl_winograd_convolution.hpp"
using namespace dnnl::impl::cpu::acl;
#elif DNNL_RV64
#include "cpu/rv64/rvv_gemm_convolution.hpp"
using namespace dnnl::impl::cpu::rv64;
#endif

namespace dnnl {
Expand Down Expand Up @@ -175,6 +178,10 @@ const std::map<pk_dt_impl_key_t, std::vector<impl_list_item_t>> &impl_list_map()
CPU_INSTANCE_AARCH64(brgemm_1x1_convolution_fwd_t, sve_128)
CPU_INSTANCE_AARCH64(brgemm_convolution_fwd_t, sve_128)
// CPU_INSTANCE_X64(jit_uni_ncsp_convolution_fwd_t)

CPU_INSTANCE_RV64GCV(riscv_gemm_convolution_fwd_t)


CPU_INSTANCE(gemm_convolution_fwd_t)
CPU_INSTANCE(ref_convolution_fwd_t)
CPU_INSTANCE(ref_fused_convolution_fwd_t)
Expand Down
Loading