Skip to content

Commit 2e996e0

Browse files
georgehongfacebook-github-bot
authored andcommitted
Update training module to have super class methods accessible
Summary: This is needed so the training module has access to non-training methods (e.g. constant string return methods). Differential Revision: D66419247
1 parent 30d5a11 commit 2e996e0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/training/module/training_module.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ namespace training {
2626
* A facade class for loading programs for on-device training and executing
2727
* methods within them.
2828
*/
29-
class ET_EXPERIMENTAL TrainingModule final : executorch::extension::Module {
29+
class ET_EXPERIMENTAL TrainingModule final : public executorch::extension::Module {
3030
public:
3131
explicit TrainingModule(
3232
std::unique_ptr<runtime::DataLoader> data_loader,

0 commit comments

Comments
 (0)