Skip to content

Commit 30d6a72

Browse files
committed
issue/594 - cumulating total time in tests
1 parent 5c88cbb commit 30d6a72

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

test/infinicore/ops/elu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,9 +133,9 @@ def torch_operator(self, *args, **kwargs):
133133
"""PyTorch ELU implementation"""
134134
return torch.nn.functional.elu(*args, **kwargs)
135135

136-
def infinicore_operator(self, x, alpha=1.0, out=None, **kwargs):
137-
"""InfiniCore ELU implementation"""
138-
return None
136+
# def infinicore_operator(self, x, alpha=1.0, out=None, **kwargs):
137+
# """InfiniCore ELU implementation"""
138+
# return None
139139

140140

141141
def main():

test/infinicore/ops/multi_margin_loss.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def parse_test_cases():
103103
return test_cases
104104

105105

106-
class MultiMarginLossOpTest(BaseOperatorTest):
106+
class OpTest(BaseOperatorTest):
107107
"""MultiMarginLoss operator test with device handling"""
108108

109109
def __init__(self):
@@ -116,9 +116,9 @@ def torch_operator(self, *args, **kwargs):
116116
"""PyTorch multi_margin_loss implementation with device handling"""
117117
return F.multi_margin_loss(*args, **kwargs)
118118

119-
def infinicore_operator(self, *args, **kwargs):
120-
"""InfiniCore multi_margin_loss implementation"""
121-
return None
119+
# def infinicore_operator(self, *args, **kwargs):
120+
# """InfiniCore multi_margin_loss implementation"""
121+
# return None
122122

123123

124124
def main():

0 commit comments

Comments
 (0)