| 1 | #pragma once |
|---|---|
| 2 | #include "common.cuh" |
| 3 | |
| 4 | #define CUDA_CONV2D_BLOCK_SIZE 256 |
| 5 | void ggml_cuda_op_conv2d(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 |
| 1 | #pragma once |
|---|---|
| 2 | #include "common.cuh" |
| 3 | |
| 4 | #define CUDA_CONV2D_BLOCK_SIZE 256 |
| 5 | void ggml_cuda_op_conv2d(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 |