Unverified Commit 1e43a28e authored by thatPepe's avatar thatPepe Committed by GitHub
Browse files

Merge pull request #589 from InfiniTensor/issue/566

issue/566: 添加 Python 算子测试
parents 798dc76e 4deec279
This diff is collapsed.
This diff is collapsed.
......@@ -56,7 +56,6 @@ def parse_test_cases():
"""
Parse ELU test case data according to format:
(shape, input_strides, alpha)
ELU only supports out-of-place and in-place modes via PyTorch's inplace parameter
"""
test_cases = []
......@@ -133,9 +132,10 @@ class OpTest(BaseOperatorTest):
"""PyTorch ELU implementation"""
return torch.nn.functional.elu(*args, **kwargs)
# def infinicore_operator(self, x, alpha=1.0, out=None, **kwargs):
# def infinicore_operator(self, *args, **kwargs):
# """InfiniCore ELU implementation"""
# return None
# return infinicore.nn.functional.elu(*args, **kwargs)
# # return None
def main():
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment