Fix weigh bias parsing and better error logging

This commit is contained in:
2024-04-20 18:36:53 +02:00
parent ecf7416f8e
commit d08567a563
6 changed files with 93 additions and 11 deletions

View File

@@ -1,6 +1,8 @@
#ifndef CUDANET_ACTIVATION_FUNCTIONS_H
#define CUDANET_ACTIVATION_FUNCTIONS_H
#include <cuda_runtime.h>
namespace CUDANet::Kernels {
/**

View File

@@ -1,6 +1,8 @@
#ifndef CUDANET_CONVOLUTION_H
#define CUDANET_CONVOLUTION_H
#include <cuda_runtime.h>
namespace CUDANet::Kernels {
/**

View File

@@ -1,6 +1,8 @@
#ifndef CUDANET_MATMUL_H
#define CUDANET_MATMUL_H
#include <cuda_runtime.h>
namespace CUDANet::Kernels {
/**

View File

@@ -4,6 +4,7 @@
#include <vector>
#include "layer.cuh"
#include "activation.cuh"
namespace CUDANet::Layers {