16 Commits

Author	SHA1	Message	Date
LordMathis	7c48ed86d2	Implement vector variance function	2024-05-14 21:58:23 +02:00
LordMathis	da8f3167cb	Add utils vector mean function	2024-05-14 21:41:18 +02:00
LordMathis	5c8d3f7e25	Compute mean and variance	2024-04-29 20:55:11 +02:00
LordMathis	0ab623fa23	Implement vector mean calculation	2024-04-28 22:04:15 +02:00
LordMathis	3320f610db	Add imagenet class map	2024-04-28 19:03:51 +02:00
LordMathis	4b9d123e94	Implement device vector utils	2024-04-11 22:22:33 +02:00
LordMathis	dc86cddeb7	Use tiling shmem for mat vec mul kernel	2024-03-15 23:33:09 +01:00
LordMathis	88f7fff217	Add prefix to guards	2024-03-13 22:23:23 +01:00
LordMathis	7157a27e56	Add documentation comments	2024-03-12 21:50:06 +01:00
LordMathis	9d91896f13	Change forward function to return output pointer	2024-03-12 20:50:49 +01:00
LordMathis	f4257afd5a	Remove cublas dependency	2024-03-05 18:41:35 +01:00
LordMathis	6e99525ad0	Rename hheader files to .cuh	2024-02-26 19:53:46 +01:00
LordMathis	035f3b053b	Rename files to .cu and fix IDX2C usage	2024-02-21 20:03:04 +01:00
LordMathis	22e33a395b	Initialize and use cuBLAS properly	2024-02-10 17:17:07 +01:00
LordMathis	3e6b7dc7e6	Implement to_cuda function	2024-02-09 19:18:30 +01:00
LordMathis	b16ec69469	Initial CUDA test	2024-02-07 21:14:46 +01:00