cuDFOnRayFrameAxisPartition¶
The class is a base class for any axis partition class based on Ray engine and cuDF backend. This class provides the API to perform operations on an axis partition, using Ray as the execution engine. The axis partition is made up of list of block partitions that are stored in this class.
Public API¶
- class modin.engines.ray.cudf_on_ray.frame.axis_partition.cuDFOnRayFrameAxisPartition(partitions)¶
Base class for any axis partition class for cuDF backend.
- Parameters
partitions (np.ndarray) – NumPy array with
cuDFOnRayFramePartition
-s.
- partition_type¶
alias of
modin.engines.ray.cudf_on_ray.frame.partition.cuDFOnRayFramePartition
cuOnRayFrameColumnPartition¶
Public API¶
- class modin.engines.ray.cudf_on_ray.frame.axis_partition.cuDFOnRayFrameColumnPartition(partitions)¶
The column partition implementation of
cuDFOnRayFrameAxisPartition
.- Parameters
partitions (np.ndarray) – NumPy array with
cuDFOnRayFramePartition
-s.
- reduce(func)¶
Reduce partitions along self.axis and apply func.
- Parameters
func (callable) – A func to apply.
- Returns
- Return type
cuDFOnRayFrameRowPartition¶
Public API¶
- class modin.engines.ray.cudf_on_ray.frame.axis_partition.cuDFOnRayFrameRowPartition(partitions)¶
The row partition implementation of
cuDFOnRayFrameAxisPartition
.- Parameters
partitions (np.ndarray) – NumPy array with
cuDFOnRayFramePartition
-s.
- reduce(func)¶
Reduce partitions along self.axis and apply func.
- Parameters
func (calalble) – A func to apply.
- Returns
- Return type
Notes
Since we are using row partitions, we can bypass the Ray plasma store during axis reduction functions.