forked from jiuyuan/InfiniTensor
48847958d0
* impl sqrt on CUDA fix parser of Gather and ReduceMean * fix test_gather * fix test_cuda_gather * impl sqrt cpu and add sqrt to test_cuda_unary * cuda_unary supports arbitary shapes * fix SplitOp with dim=-1 * fix SplitOp with dim=-1 |
||
---|---|---|
.. | ||
docs | ||
src/pyinfinitensor | ||
tests | ||
pyproject.toml |