Migrate output layer

2025-11-06 09:44:28 +00:00 · 2024-09-10 19:20:00 +02:00
parent f7b525e494
commit 74f49d6a00
6 changed files with 34 additions and 12 deletions
--- a/test/cuda/layers/test_output.cu
+++ b/test/cuda/layers/test_output.cu
@@ -1,7 +1,7 @@
 #include <cuda_runtime_api.h>
 #include <gtest/gtest.h>

-#include "output.cuh"
+#include "output.hpp"

 TEST(OutputLayerTest, OutputForward) {
    cudaError_t cudaStatus;