Skip to content
Snippets Groups Projects
Commit 3ad39df8 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fix typos

parent 0f71ab1d
No related branches found
No related tags found
1 merge request!14version 0.1.1
...@@ -250,11 +250,11 @@ class AddCPP(ExportNode): ...@@ -250,11 +250,11 @@ class AddCPP(ExportNode):
def forward(self, list_actions:list): def forward(self, list_actions:list):
list_actions.append(set_up_output(self.name, "float")) list_actions.append(set_up_output(self.name, "float"))
list_actions.append(generate_action( list_actions.append(generate_str(
str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"), str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"),
name=self.name, name=self.name,
inputs1_name=self.parents[0].name(), inputs1_name=self.inputs[0].name(),
inputs2_name=self.parents[1].name(), inputs2_name=self.inputs[1].name(),
output_name=self.name output_name=self.name
)) ))
return list_actions return list_actions
...@@ -283,7 +283,7 @@ class SubCPP(ExportNode): ...@@ -283,7 +283,7 @@ class SubCPP(ExportNode):
def forward(self, list_actions:list): def forward(self, list_actions:list):
list_actions.append(set_up_output(self.name, "float")) list_actions.append(set_up_output(self.name, "float"))
list_actions.append(generate_action( list_actions.append(generate_str(
str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"), str(ROOT / "templates" / "kernel_forward" / "elemwise_forward.jinja"),
name=self.name, name=self.name,
inputs1_name=self.inputs[0].name(), inputs1_name=self.inputs[0].name(),
......
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