search

NVIDIA / Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
Other
271 stars 53 forks source link

Use IterDomain::split and IterDomain::merge #3327

Closed naoyam closed 2 weeks ago

naoyam commented 2 weeks ago

Instead of manually creating output IDs and directly constructing Split and Merge nodes, IterDomain::split and IterDomain::merge are used. This would reduce duplication of logic.

There's some slight change in iteration types of IDs generated when TensorView::rFactor is used. See the inline comment.

As of 8321cfd, everything seems fine.

Closes #2520

naoyam commented 2 weeks ago

!test --diff

naoyam commented 2 weeks ago

!test --diff

naoyam commented 2 weeks ago

!build

naoyam commented 2 weeks ago

!test

naoyam commented 2 weeks ago

!build

xwang233 commented 2 weeks ago

!build