search

cornell-zhang / heterocl

HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing
https://cornell-zhang.github.io/heterocl/
Apache License 2.0
326 stars 92 forks source link

[API] Fix hcl.print with Incorrect Output Shape #409

Closed seanlatias closed 3 years ago

seanlatias commented 3 years ago

Fixed issue: #408

Detailed description: In this PR, we make sure the output shape is the same as the input shape. The bug was from incorrect loop bound (tensor shape) collection when printing the tensor.

Link to the tests: tests/issues/test_issue_408.py A separate test is also added under tests/test_api_print.py