slice_scatter decomposition by apbose · Pull Request #2519 · pytorch/TensorRT

@apbose apbose marked this pull request as draft

December 6, 2023 09:08

github-actions[bot]

github-actions[bot]

github-actions[bot]

github-actions[bot]

github-actions[bot]

github-actions[bot]

@apbose apbose marked this pull request as ready for review

January 2, 2024 22:00

github-actions[bot]

github-actions[bot]

gs-olive

gs-olive

gs-olive

gs-olive

gs-olive

gs-olive

github-actions[bot]

github-actions[bot]

gs-olive

gs-olive

@apbose

changing decomposition pattern

slice scatter changes

Review comments address

Removing arange and replacing with range

slice_scatter adding to decomposition group

using aten::scatter in aten.slice_scatter

Correcting the slice_scatter case with aten::scatter use

removing unnecessary cases from slice_scatter impl and adding test case

changing for loop to torch.arange

Reverting back the torch.arange to for loop

Adding test case for 3d cases and removing the casting to torch.int64 and including it torch.ones

Removing aten.index in the decomposition ops