blob: 36e56fe8986f336acc6d257b49c44b537215604d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
#pragma once
#include "ParallelForRunner.h"
#include "Tensor.h"
namespace CpuCompute
{
HRESULT mulMat( Tensor& result, const Tensor& a, const Tensor& b, ParallelForRunner& pfor );
}
#if TENSOR_GGML_COMPAT
#include "../source/ggml.h"
inline HRESULT mulMat( ggml_tensor* result, const ggml_tensor* a, const ggml_tensor* b, CpuCompute::ParallelForRunner& pfor )
{
CpuCompute::Tensor r{ result }, lhs{ a }, rhs{ b };
return CpuCompute::mulMat( r, lhs, rhs, pfor );
}
#endif
|
