soklyn 0.1.0

A high-performance neural network library written in Rust, utilizing CUDA for GPU-accelerated layer execution.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

cmake_minimum_required(VERSION 3.18)
project(cuda_nn_math LANGUAGES CXX CUDA)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CUDA_STANDARD 17)

if(CMAKE_EXPORT_COMPILE_COMMANDS)
    add_compile_definitions(__CUDA_ARCH__=800)
endif()

add_library(nn_math OBJECT nn_math.cu
        kernel/types.hpp
        kernel/util.cu kernel/math.cu
        kernel/ffn/forward.cu kernel/ffn/backward.cu
        kernel/ffn/forward_wmma.cu kernel/ffn/backward_wmma.cu)

set_target_properties(
        nn_math
        PROPERTIES
        CUDA_SEPARABLE_COMPILATION ON
)