pytorch
20d2260e - FX Partitioner: use "all aten ops that decompose to prims" as supported NVFuser ops

Commit
2 years ago
FX Partitioner: use "all aten ops that decompose to prims" as supported NVFuser ops Previously we were using the list taken from the torchscript integration. But in this case it's actually more useful to just use the list of ops that can decompose to prims, because
Author
Committer
Parents
Loading