geocat-comp icon indicating copy to clipboard operation
geocat-comp copied to clipboard

Relhum function on GPU using CuPy

Open haniyeka opened this issue 2 years ago • 7 comments

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.

haniyeka avatar Jul 08 '22 20:07 haniyeka