aboutsummaryrefslogtreecommitdiffstats
path: root/modules/devices.py
Commit message (Expand)AuthorAgeFilesLines
* Updatewangshuai092024-01-311-0/+7
* Merge branch 'dev' into npu_supportwangshuai092024-01-301-3/+95
|\
| * Revert "Try to reverse the dtype checking mechanism"Kohaku-Blueleaf2024-01-291-2/+5
| * Try to reverse the dtype checking mechanismKohaku-Blueleaf2024-01-291-5/+2
| * lintingKohaku-Blueleaf2024-01-291-1/+0
| * Fix potential bugsKohaku-Blueleaf2024-01-291-2/+7
| * Avoid exceptions to be silencedKohaku-Blueleaf2024-01-201-6/+5
| * Avoid early disableKohaku-Blueleaf2024-01-201-0/+4
| * Fix nested manual castKohaku-Blueleaf2024-01-181-1/+5
| * rearrange if-statements for cpuKohaku-Blueleaf2024-01-091-3/+3
| * Apply the correct behavior of precision='full'Kohaku-Blueleaf2024-01-091-4/+7
| * Revert "Apply correct inference precision implementation"Kohaku-Blueleaf2024-01-091-33/+9
| * Apply correct inference precision implementationKohaku-Blueleaf2024-01-091-9/+33
| * linting and debugsKohaku-Blueleaf2024-01-091-6/+6
| * Fix bugs when arg dtype doesn't matchKohakuBlueleaf2024-01-091-15/+10
| * improve efficiency and support more deviceKohaku-Blueleaf2024-01-091-17/+43
| * change import statements for #14478AUTOMATIC11112023-12-311-2/+2
| * Add utility to inspect a model's parameters (to get dtype/device)Aarni Koskela2023-12-311-1/+2
| * Merge branch 'dev' into test-fp8Kohaku-Blueleaf2023-12-031-0/+13
| |\
| * \ Merge branch 'dev' into test-fp8Kohaku-Blueleaf2023-12-021-1/+1
| |\ \
| * | | Better namingKohaku-Blueleaf2023-11-191-3/+3
| * | | Use options instead of cmd_argsKohaku-Blueleaf2023-11-191-11/+14
| * | | Add MPS manual castKohakuBlueleaf2023-10-281-1/+5
| * | | ManualCast for 10/16 series gpuKohaku-Blueleaf2023-10-281-6/+51
| * | | Add CPU fp8 supportKohaku-Blueleaf2023-10-231-1/+5
* | | | Add NPU Supportwangshuai092024-01-291-2/+7
| |_|/ |/| |
* | | Merge pull request #14171 from Nuullll/ipexAUTOMATIC11112023-12-021-0/+13
|\ \ \ | |_|/ |/| |
| * | Disable ipex autocast due to its bad perfNuullll2023-12-021-7/+13
| * | Initial IPEX supportNuullll2023-11-301-2/+9
| |/
* | Merge pull request #14131 from read-0nly/patch-1AUTOMATIC11112023-12-021-1/+1
|\ \ | |/ |/|
| * Update devices.pyobsol2023-11-281-1/+1
* | fix for crash when running #12924 without --device-idAUTOMATIC11112023-09-091-1/+1
* | More accurate check for enabling cuDNN benchmark on 16XX cardscatboxanon2023-08-311-1/+2
|/
* split shared.py into multiple files; should resolve all circular reference im...AUTOMATIC11112023-08-091-9/+1
* rework RNG to use generators instead of generating noises beforehandAUTOMATIC11112023-08-091-79/+2
* rework torchsde._brownian.brownian_interval replacement to use device.randn_l...AUTOMATIC11112023-08-031-6/+38
* add NV option for Random number generator source setting, which allows to gen...AUTOMATIC11112023-08-021-2/+37
* Fix MPS cache cleanupAarni Koskela2023-07-111-2/+3
* added torch.mps.empty_cache() to torch_gc()AUTOMATIC11112023-07-081-0/+3
* Remove a bunch of unused/vestigial codeAarni Koskela2023-06-051-7/+0
* run basic torch calculation at startup in parallel to reduce the performance ...AUTOMATIC2023-05-211-0/+18
* ruff auto fixesAUTOMATIC2023-05-101-1/+1
* rename CPU RNG to RNG source in settings, add infotext and parameters copypas...AUTOMATIC2023-04-291-2/+2
* Option to use CPU for random number generation.Deciare2023-04-191-2/+6
* Refactor Mac specific code to a separate filebrkirch2023-02-011-45/+7
* Refactor MPS fixes to CondFuncbrkirch2023-02-011-36/+14
* MPS fix is still needed :(brkirch2023-02-011-0/+3
* Merge pull request #7309 from brkirch/fix-embeddingsAUTOMATIC11112023-01-281-3/+8
|\
| * Remove MPS fix no longer needed for PyTorchbrkirch2023-01-281-3/+0
| * Refactor conditional casting, fix upscalersbrkirch2023-01-281-0/+8