blob: c30128561e810b169527eb939e662d8c95c4f912 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
#pragma once
#include "ggml-cuda/common.cuh"
#include "ggml.h"
// Asynchronously copies data from src tensor to dst tensor using the provided context.
// Returns a musaError_t indicating success or failure.
musaError_t mudnnMemcpyAsync(
ggml_backend_cuda_context &ctx,
const ggml_tensor *dst,
const ggml_tensor *src
);
|
