lancer
|
5c90845f7a
|
update the -lcublas -lcublasLt flag in the comment
|
2024-05-09 22:04:02 -07:00 |
|
Erik Schultheis
|
06980b6b2f
|
mixed precision utilities for dev/cuda
|
2024-05-01 22:35:55 +03:00 |
|
lancer
|
3cb993b5f6
|
fix the constness and the warning C4305: 'initializing': truncation from 'double' to 'float'
|
2024-04-19 11:04:58 -07:00 |
|
Andrej Karpathy
|
d74d7ae652
|
small adamw fixes, needs more work
|
2024-04-17 04:58:16 +00:00 |
|
nopperl
|
97e61e2258
|
add slightly more optimized adamw kernel
|
2024-04-17 03:31:07 +02:00 |
|
nopperl
|
5684ff7e77
|
add adamw cuda kernel
|
2024-04-17 03:30:59 +02:00 |
|