geocat-comp
geocat-comp copied to clipboard
Relhum function on GPU using CuPy
geocat_cupy.yml: Created an export from the geocat Conda environment, including the required packages.
meteorology.py: The Relhum function and its helper functions _relhum, and _xrelhum are ported to do the computation on GPU using CuPy. They have a new input variable, "use_gpu", which its default value is set to "False", but when the user sets it to "True", the computation of the Relhum function will be done on GPU.
test_meteorology.py: The Test_Relhum class is changed according to the changes in meteolorogy.py. They also have a new input variable, the "use_gpu", which is set to "False" by default.
Note: The current ported version works on a single GPU device, but the next step is to try running the function on multiple GPUs.