Adjusting the header file so that I can use it for opencl as well.