Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
396e07b
causal_conv1d: use PTO-ISA kernel implementation
ChristosMatzoros Jun 23, 2026
549e4ba
causal_conv1d: support widths 2-64, fix writeback race, clang-format …
ChristosMatzoros Jun 23, 2026
3dabb93
causal_conv1d: template on ring size RS, take width K at runtime
ChristosMatzoros Jun 24, 2026
c5e1757
causal_conv1d: cast weight/bias to fp32 in the kernel
ChristosMatzoros Jun 25, 2026
87344fe
causal_conv1d: minimize casts via a signed index space
ChristosMatzoros Jun 25, 2026
2a04712
causal_conv1d: apply clang-format
ChristosMatzoros Jun 25, 2026
d31fb09
causal_conv1d: larger max width for tap length of 2
raphael-s-steiner Jun 25, 2026
d83c691
different tiling
raphael-s-steiner Jul 14, 2026
c625abc
wrong sign
raphael-s-steiner Jul 15, 2026
3a10103
avoiding zero
raphael-s-steiner Jul 15, 2026
84e47a2
slightly improved
raphael-s-steiner Jul 15, 2026
5161ef7
improved tiling
raphael-s-steiner Jul 15, 2026
0fb38a8
faster tile parameters
raphael-s-steiner Jul 15, 2026
9c376dc
improved tiling
raphael-s-steiner Jul 15, 2026
106e88d
typo
raphael-s-steiner Jul 15, 2026
0106dfb
format
raphael-s-steiner Jul 15, 2026
6b3e05b
minor improvement
raphael-s-steiner Jul 16, 2026
13ad3a2
causal_conv1d: register the PTO op schema (drop #592 run_mode/num_acc…
ChristosMatzoros Jul 20, 2026
4acf6df
causal_conv1d: address review comments
ChristosMatzoros Jul 22, 2026
c279fc9
causal_conv1d: support A5
ChristosMatzoros Jul 30, 2026
a9a5aff
causal_conv1d: guard the A5 tile table and UB size on __DAV_C310__
ChristosMatzoros Jul 31, 2026
3d0aa11
causal_conv1d: fuse the scatter accumulate with TMULADDDST on A5
ChristosMatzoros Aug 6, 2026
7e7aafe
causal_conv1d: drop the inline comments
ChristosMatzoros Aug 6, 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
14 changes: 13 additions & 1 deletion csrc/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,6 @@ set(WORKSPACE_KERNEL_SRCS
${PROJECT_OP_SRC_BASE}/build_tree/op_kernel/build_tree_kernel.cpp
${PROJECT_OP_SRC_BASE}/lightning_indexer/op_kernel/lightning_indexer_kernel.cpp
${PROJECT_OP_SRC_BASE}/causal_conv1d_update/op_kernel/causal_conv1d_update.cpp
${PROJECT_OP_SRC_BASE}/causal_conv1d/op_kernel/causal_conv1d.cpp
${PROJECT_OP_SRC_BASE}/lora/op_kernel/sgemmc_expand_kernel.cpp
${PROJECT_OP_SRC_BASE}/lora/op_kernel/sgemmc_shrink_kernel.cpp
)
Expand Down Expand Up @@ -132,11 +131,24 @@ ascendc_compile_definitions(mega_chunk_gdn_kernel PRIVATE
-DGDN_C=128
)

# PTO-ISA causal_conv1d kernel (depthwise conv + bias + SiLU), built like mega_chunk_gdn.
ascendc_library(causal_conv1d_kernel STATIC
${PROJECT_OP_SRC_BASE}/causal_conv1d/op_kernel/causal_conv1d.cpp
)
ascendc_include_directories(causal_conv1d_kernel PRIVATE
${PROJECT_OP_SRC_BASE}/causal_conv1d/op_kernel
${PROJECT_SOURCE_DIR}/third_party/pto-isa/include
${ASCEND_INCLUDE_DIR}
${ASCEND_INCLUDE_DIR}/experiment/runtime
${ASCEND_INCLUDE_DIR}/experiment/msprof
)

# create shared library libsgl_kernel_npu.so
add_library(${OP_PLUGIN_NAME} SHARED ${OP_SRCS})

target_link_libraries(${OP_PLUGIN_NAME} PRIVATE
${MEGA_CHUNK_GDN_KERNEL_TARGETS}
causal_conv1d_kernel
workspace_kernel
no_workspace_kernel
torch_npu
Expand Down
Loading
Loading