Skip to content

Commit

Permalink
[RunLayerContext] Unused property and its relevant methods are removed
Browse files Browse the repository at this point in the history
- This commit removes unused property and its relevant methods in
  RunLayerContext.
- compute_engine info is not handled in RunLayerContext.

Signed-off-by: Eunju Yang <[email protected]>
  • Loading branch information
EunjuYang committed Dec 31, 2024
1 parent 24a868d commit 079f57a
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions nntrainer/layers/layer_context.h
Original file line number Diff line number Diff line change
Expand Up @@ -901,21 +901,6 @@ class RunLayerContext {
*/
std::vector<Weight *> 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
Expand Down

0 comments on commit 079f57a

Please sign in to comment.