Change forward function to return output pointer

This commit is contained in:
2024-03-12 20:50:49 +01:00
parent 2518138ef8
commit 9d91896f13
8 changed files with 69 additions and 113 deletions

View File

@@ -2,6 +2,7 @@
#define CUDA_HELPER_H
#include <cuda_runtime.h>
#include <cstdio>
// CUDA error checking macro
#define CUDA_CHECK(call) \