ACloudViewer  3.9.4
A Modern Library for 3D Data Processing
Matmul.h
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // - CloudViewer: www.cloudViewer.org -
3 // ----------------------------------------------------------------------------
4 // Copyright (c) 2018-2024 www.cloudViewer.org
5 // SPDX-License-Identifier: MIT
6 // ----------------------------------------------------------------------------
7 
8 #pragma once
9 
10 #include "cloudViewer/core/Tensor.h"
11 
12 namespace cloudViewer {
13 namespace core {
14 
16 void Matmul(const Tensor& A, const Tensor& B, Tensor& C);
17 
18 #ifdef BUILD_SYCL_MODULE
19 void MatmulSYCL(void* A_data,
20  void* B_data,
21  void* C_data,
22  int64_t m,
23  int64_t k,
24  int64_t n,
25  Dtype dtype,
26  const Device& device);
27 #endif
28 #ifdef BUILD_CUDA_MODULE
29 void MatmulCUDA(void* A_data,
30  void* B_data,
31  void* C_data,
32  int64_t m,
33  int64_t k,
34  int64_t n,
35  Dtype dtype,
36  const Device& device);
37 #endif
38 void MatmulCPU(void* A_data,
39  void* B_data,
40  void* C_data,
41  int64_t m,
42  int64_t k,
43  int64_t n,
44  Dtype dtype);
45 } // namespace core
46 } // namespace cloudViewer
cloudViewer::core::MatmulCUDA
void MatmulCUDA(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype, const Device &device)
Definition: MatmulCUDA.cpp:17
cloudViewer::core::MatmulSYCL
void MatmulSYCL(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype, const Device &device)
Definition: MatmulSYCL.cpp:19
cloudViewer::core::Matmul
void Matmul(const Tensor &A, const Tensor &B, Tensor &output)
Computes matrix multiplication C = AB.
Definition: Matmul.cpp:17
cloudViewer::core::MatmulCPU
void MatmulCPU(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype)
Definition: MatmulCPU.cpp:14
cloudViewer
Generic file read and write utility for python interface.
Definition: AutoSegmentationTools.h:16