Issue #27

modin/config/__main__.py

@@ -19,6 +19,7 @@ def print_config_help():
     for objname in sorted(globals()):
         obj = globals()[objname]
         if isinstance(obj, type) and issubclass(obj, Parameter) and not obj.is_abstract:
+            pring("print_config_help()")
             print(f"{obj.get_help()}\n\tCurrent value: {obj.get()}")  # noqa: T001
 
 

modin/config/envvars.py

@@ -94,7 +94,7 @@ class Backend(EnvironmentVariable, type=str):
 
     varname = "MODIN_BACKEND"
     default = "Pandas"
-    choices = ("Pandas", "OmniSci", "Pyarrow")
+    choices = ("Pandas", "OmniSci", "Pyarrow", "cuDF")
 
 
 class IsExperimental(EnvironmentVariable, type=bool):
@@ -135,6 +135,26 @@ def _get_default(cls):
         return multiprocessing.cpu_count()
 
 
+class GpuCount(EnvironmentVariable, type=int):
+    """
+    How may GPU devices to utilize across the whole distribution
+    """
+
+    varname = "MODIN_GPUS"
+
+    # @classmethod
+    # def _get_default(cls):
+    #     import ray
+    #     if ray.is_initialized():
+    #         return len(ray.get_gpu_ids())
+    #     else:
+    #         print(
+    #             f"Ray has not been initialized yet. "
+    #             f"MODIN-GPU doesn't know the number of available GPUs. "
+    #         )
+    #         return 1
+
+
 class Memory(EnvironmentVariable, type=int):
     """
     How much memory give to each Ray worker (in bytes)
@@ -152,7 +172,10 @@ class NPartitions(EnvironmentVariable, type=int):
 
     @classmethod
     def _get_default(cls):
-        return CpuCount.get()
+        if Backend.get() == "cuDF":
+            return GpuCount.get()
+        else:
+            return CpuCount.get()
 
 
 class RayPlasmaDir(EnvironmentVariable, type=ExactStr):

modin/pandas/__init__.py

@@ -100,7 +100,7 @@
 
 def _update_engine(publisher: Parameter):
     global dask_client
-    from modin.config import Backend, CpuCount
+    from modin.config import Backend, CpuCount, GpuCount
 
     if publisher.get() == "Ray":
         from modin.engines.ray.utils import initialize_ray