From 6e383f49a777236cec9772c88faf32a7cc5c8c1f Mon Sep 17 00:00:00 2001 From: Eunju Yang Date: Tue, 31 Dec 2024 12:47:16 +0900 Subject: [PATCH] [RunLayerContext] Unused property and its relevant methods are removed - This commit removes unused property and its relevant methods in RunLayerContext. - compute_engine info is not handled in RunLayerContext. Signed-off-by: Eunju Yang --- nntrainer/layers/layer_context.h | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h index fef7ff6d7..33839a143 100644 --- a/nntrainer/layers/layer_context.h +++ b/nntrainer/layers/layer_context.h @@ -901,21 +901,6 @@ class RunLayerContext { */ std::vector getWeights() { return weights; } - /** - * @brief set the compute engine for this node - * @param compute engine: (CPU/GPU) - */ - void setComputeEngine(const ml::train::LayerComputeEngine &compute_engine = - ml::train::LayerComputeEngine::CPU) { - this->compute_engine = compute_engine; - } - - /** - * @brief get the compute engine for this node - * @return ompute engine: (CPU/GPU) - */ - ml::train::LayerComputeEngine getComputeEngine() { return compute_engine; } - /** * @brief get loss scale * @return loss scale @@ -958,9 +943,6 @@ class RunLayerContext { std::vector outputs; /**< outputs of the layer */ std::vector tensors; /**< tensors of the layer */ - ml::train::LayerComputeEngine compute_engine = - ml::train::LayerComputeEngine::CPU; - #ifdef DEBUG std::map tensor_map; /**< map of tensor name to tensor address */