File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -32,7 +32,8 @@ file(GLOB_RECURSE CLIENT_SOURCES "src/client/*.cpp")
3232# include(FetchContent)
3333
3434find_package (CUDAToolkit REQUIRED COMPONENTS cublas )
35- set (CUDA_TOOLKIT_ROOT_DIR /usr/local/cuda-12.6) # 根据您的安装路径进行调整
35+ include_directories (${CUDAToolkit_INCLUDE_DIRS} )
36+
3637set (CMAKE_CUDA_STANDARD 17)
3738set (CMAKE_CUDA_ARCHITECTURES 61) # 根据您的 GPU 计算能力进行调整
3839set (CMAKE_CUDA_SEPARABLE_COMPILATION ON ) #确保 CMake 能够正确识别 CUDA 文件并将其编译为目标
Original file line number Diff line number Diff line change 11#ifndef DEEPX_TENSORFUNC_EQUAL_HPP
22#define DEEPX_TENSORFUNC_EQUAL_HPP
33#include < cmath>
4- #include < omp.h>
5-
4+
65#include " deepx/tensor.hpp"
76#include " deepx/shape.hpp"
87namespace deepx ::tensorfunc
Original file line number Diff line number Diff line change 11#ifndef DEEPX_TENSORFUNC_MATMUL_HPP
22#define DEEPX_TENSORFUNC_MATMUL_HPP
3-
4- #include < cblas.h> // 如果使用 OpenBLAS
3+
54#include " deepx/tensor.hpp"
65
76namespace deepx ::tensorfunc
Original file line number Diff line number Diff line change 55#include < algorithm>
66#include < stdexcept>
77#include < deepx/vector_combination.hpp>
8- #include < hwy/highway.h>
98
109#include " deepx/tensor.hpp"
1110#include " deepx/shape_reduce.hpp"
You can’t perform that action at this time.
0 commit comments