Source code for nv_ingest.framework.orchestration.morpheus.stages.meta.linear_module_source_stage_cpu

# SPDX-FileCopyrightText: Copyright (c) 2024-25, NVIDIA CORPORATION & AFFILIATES.
# All rights reserved.
# SPDX-License-Identifier: Apache-2.0

from morpheus.config import ExecutionMode
from morpheus.stages.general.linear_modules_source import LinearModuleSourceStage
from morpheus.stages.general.linear_modules_stage import LinearModulesStage



[docs]
class LinearModuleSourceStageCPU(LinearModuleSourceStage):

[docs]
    def supported_execution_modes(self) -> tuple[ExecutionMode]:
        # Provide your own logic here; for example:
        return (ExecutionMode.CPU,)





[docs]
class LinearModuleStageCPU(LinearModulesStage):

[docs]
    def supported_execution_modes(self) -> tuple[ExecutionMode]:
        # Provide your own logic here; for example:
        return (ExecutionMode.CPU,)