oskar-inceptron opened a new pull request, #17688:
URL: https://github.com/apache/tvm/pull/17688
The current code asserts that floormod(input_dims[0], num_shards) can not be
proven to be true, which has two problems:
- It is unclear what it means to prove floormod(..). Prove that floormod's
return value is truthy, i.e. non-zero?
- It always checks the 0th dimension of the input shape, but the dimension
index to be sharded is given by the "axis" parameter.
This commit fixes both of the above.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
