Skip to content
Snippets Groups Projects
Commit d66d2c43 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Added repr for torchapp

parent 92a20e60
No related branches found
No related tags found
No related merge requests found
......@@ -124,6 +124,14 @@ class TorchApp(ModeledApp, abc.ABC):
qos = self.combine_qos(np.array(qoses))
return 0.0, qos
def __repr__(self) -> str:
class_name = self.__class__.__name__
module_class_name = type(self.module).__name__
return (
f'{class_name}"{self.name}"(module={module_class_name}, '
f"num_op={len(self._op_knobs)}, num_knob={len(self.name_to_knob)})"
)
@torch.no_grad()
def _get_raw_output_valset(self, with_approxes: KnobsT):
approxed = self._apply_knobs(with_approxes)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment