search

nod-ai / sharktank

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

Expand sharded element-wise binary ops support #63

Closed sogartar closed 3 weeks ago

sogartar commented 3 weeks ago

Add polymorphic replicate, shard and shard_like ops. Add operator + for InferenceTensor. Add ops.equal, analogous to torch.equal that operates on sharktank tensor types.

sogartar commented 3 weeks ago

I would actually like to add a few more tests.