Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Beinsezii's avatar
    UniPC Multistep fix tensor dtype/device on order=3 (#7532) · 19ab04ff
    Beinsezii authored 
    * UniPC UTs iterate solvers on FP16

It wasn't catching errs on order==3. Might be excessive?

* UniPC Multistep fix tensor dtype/device on order=3

* UniPC UTs Add v_pred to fp16 test iter

For completions sake. Probably overkill?
    19ab04ff
scheduling_unipc_multistep.py 37.8 KB