CUDA: add conv_2d_dw (#14265)

* CUDA: add conv_2d_dw

* better naming

* simplify using template

* Review: fix operation ordering in ggml-cuda, use __forceinline__, use more const
This commit is contained in:
Aman Gupta
2025-06-20 09:50:24 +08:00
committed by GitHub
parent 8f71d0f3e8
commit 9eaa51e7f0
3 changed files with 171 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
#pragma once
#include "common.cuh"
#define CUDA_CONV2D_DW_BLOCK_SIZE 256
void ggml_cuda_op_conv2d_dw(ggml_backend_cuda_context & ctx, ggml_tensor * dst);