daisytuner / docc / 29989607664

64%

Build Type

push

github

Committed by web-flow

Commit Message

Add tensor concatenation to PyTorch frontend (and MLIR frontend) (#868)

- Added a concat tensor node that concatenates arbitrarily many tensors
- Mapped `torch.ops.aten.cat.default` to concat tensor node
- Changed translation of `tensor.concat` MLIR op to concat tensor node instead of loop nest
- Added replace_symbols method to all types to replace symbols in types like the tensor type
- Added target specific expansion for concat tensor node to avoid branching on GPUs

Coverage Stats

352 of 454 new or added lines in 19 files covered. (77.53%)

1 existing line in 1 file now uncovered.

42653 of 66561 relevant lines covered (64.08%)

739.89 hits per line