Use 'float32' to force execution on GPU