diff --git a/code_to_optimize/code_directories/simple_tracer_e2e/workload.py b/code_to_optimize/code_directories/simple_tracer_e2e/workload.py index 063257a23..7b8fc1923 100644 --- a/code_to_optimize/code_directories/simple_tracer_e2e/workload.py +++ b/code_to_optimize/code_directories/simple_tracer_e2e/workload.py @@ -23,10 +23,9 @@ def __init__(self, num_classes=1000): self.features_size = 256 * 6 * 6 def forward(self, x): - features = self._extract_features(x) - - output = self._classify(features) - return output + # Optimization: Since _extract_features returns [] and _classify([]) returns [], + # we directly return [] + return [] def _extract_features(self, x): # As no operation is being performed, return an empty list directly