Commit Graph

6 Commits

Author SHA1 Message Date
58af95eb25 Remove cudaDeviceReset calls from tests 2024-04-21 22:47:12 +02:00
18522c2dea Cleanup and refactor 2024-04-11 22:52:41 +02:00
710a33bdde Move softmax partial kernels to matmul 2024-04-11 22:01:47 +02:00
e419a93408 Fix softmax sum kernel 2024-04-08 22:09:18 +02:00
7bc329a043 Add more softmax tests 2024-03-22 22:32:08 +01:00
42d646750b Abstract activation and implement softmax 2024-03-17 18:37:15 +01:00