Skip to content

GitLab

Commit 77ad94a9 authored by Your Name's avatar Your Name Committed by MaYuhang
Browse files

modify stack operator test

parent fe164657
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 7 deletions
test/infinicore/ops/stack.py
View file @ 77ad94a9
...@@ -13,11 +13,6 @@ from framework.runner import GenericTestRunner ...@@ -13,11 +13,6 @@ from framework.runner import GenericTestRunner
# ============================================================================== # ==============================================================================
# Test cases format: (base_shape, num_tensors, dim) # Test cases format: (base_shape, num_tensors, dim)
#
# base_shape : shape of each input tensor
# num_tensors : number of tensors to stack
# dim : dimension at which to insert the new axis
_TEST_CASES_DATA = [ _TEST_CASES_DATA = [
# ========== Basic cases ========== # ========== Basic cases ==========
((8,), 2, 0), ((8,), 2, 0),
...@@ -84,7 +79,7 @@ def parse_test_cases(): ...@@ -84,7 +79,7 @@ def parse_test_cases():
cases.append( cases.append(
TestCase( TestCase(
inputs=input_specs, inputs=[tuple(input_specs)],
kwargs=kwargs, kwargs=kwargs,
output_spec=None, output_spec=None,
comparison_target=None, comparison_target=None,
...@@ -120,4 +115,4 @@ def main(): ...@@ -120,4 +115,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() main()
\ No newline at end of file
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