Skip to content

GitLab

"profiler/src/profile_batched_gemm_multi_d.cpp" did not exist on "016ebaa7f33d2c3e86cd617210bd636fe7c99b42"
Switch branch/tag
Find file BlameHistoryPermalink
  • Arsalan's avatar
    Device agnostic testing (#5612) · f427345a
    Arsalan authored 
    * utils and test modifications to enable device agnostic testing

* device for manual seed in unet1d

* fix generator condition in vae test

* consistency changes to testing

* make style

* add device agnostic testing changes to source and one model test

* make dtype check fns private, log cuda fp16 case

* remove dtype checks from import utils, move to testing_utils

* adding tests for most model classes and one pipeline

* fix vae import
    f427345a
test_models_unet_2d_condition.py 45.7 KB