search

nod-ai / sharktank

SHARK Inference Modeling and Serving
Apache License 2.0
7 stars 9 forks source link

Add shareded unreduced tensor type #59

Closed sogartar closed 4 weeks ago

sogartar commented 4 weeks ago

Introduces a sharded tensor type that needs to be reduced over the shards to get the actual tensor. Make matmul produce such a tensor in one case.