xla
Replicate test dynamic shapes models with the equivalent version of torch.compile
#6578
Open
Open
Replicate test dynamic shapes models with the equivalent version of torch.compile #6578
changm
wants to merge
2 commits into
master
from
changm/torch
vanbasten23
commented
on 2024-02-22
vanbasten23
commented
on 2024-02-22
vanbasten23
commented
on 2024-02-22
vanbasten23
commented
on 2024-02-22
Replicate test dynamic shapes with torch.compile interface
bb740d42
Move tensors to XLA
a014fe08
vanbasten23
approved these changes
on 2024-03-05
Login to write a write a comment.
Login via GitHub