Small tweak necessary for NVIDIA.
Working on CPU.
Working on GPU.