morpheus.pipeline.execution_mode_mixins.GpuAndCpuMixin
- class GpuAndCpuMixin[source]
Bases:
abc.ABC
Mixin intented to be added to stages which support both GPU and CPU execution modes.
- Attributes
df_type_str
Returns the DataFrame module that should be used for the given execution mode.
Methods
get_df_class
()Returns the DataFrame class that should be used for the given execution mode. get_df_pkg
()Returns the DataFrame package that should be used for the given execution mode. supported_execution_modes
()Returns a tuple of supported execution modes of this stage. - property df_type_str: Literal['cudf', 'pandas']
Returns the DataFrame module that should be used for the given execution mode.
- get_df_class()[source]
Returns the DataFrame class that should be used for the given execution mode.
- get_df_pkg()[source]
Returns the DataFrame package that should be used for the given execution mode.
- supported_execution_modes()[source]
Returns a tuple of supported execution modes of this stage.