Fix matmul and max reduce memcheck errors

This commit is contained in:
2024-04-21 22:11:02 +02:00
parent 9a6152469a
commit bdbb3f978e
4 changed files with 16 additions and 41 deletions

View File

@@ -3,6 +3,7 @@
#include <cuda_runtime.h>
#include <vector>
TEST(ActivationTest, SoftmaxTest1) {
const int inputSize = 5;
cudaError_t cudaStatus;
@@ -39,6 +40,9 @@ TEST(ActivationTest, SoftmaxTest1) {
cudaFree(d_input);
cudaDeviceReset();
cudaStatus = cudaGetLastError();
EXPECT_EQ(cudaStatus, cudaSuccess);
}
TEST(ActivationTest, SoftmaxTest2) {