GitLab

Add --disable-cuda-malloc to disable it.

This is to make it match the official checkpoint.

but faster.

The wrong dimensions were being checked, [1] and [2] are the image size.
not [2] and [3]. This results in an out-of-bounds error if one of them
actually matches.