vchiley's picture
add explicit cast where running without autocast causes issues
bce2759