apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Diffusion] Add C++ dpm solver (#714)

Browse Source 
* Add BetaForAlphaBar, ConvertModelOutput, SetTimesteps, and constructor for DPMSolverMultistepScheduler

* tmp

* Add DPMSolverFirstOrderUpdate

* Add ScaleModelInput

* Add MultiStepDPMSolverSecondOrderUpdate

* add MultiStepDPMSolverThirdOrderUpdate

* Add Step

* Add FASTDEPLOY_DECL

* Add AddNoise

* Fix operator

* update

* Fix DPMSolverMultistepScheduler

* Upgrade Slice

* Fix DPMSolverFirstOrderUpdate

* remove FASTDEPLOY_DECL

* Add config for dpm solver
This commit is contained in:
Jack Zhou
2022-11-30 13:41:22 +08:00
committed by GitHub
parent 3f8ed9bfee
commit d95094cfe5
14 changed files with 675 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/function/test_elementwise.cc
View File

@@ -164,6 +164,15 @@ TEST(fastdeploy, check_same_dim) {
check_shape(z.shape, {2, 3, 4});
check_data(reinterpret_cast<const float*>(z.Data()), maximum_result.data(),
maximum_result.size());
x = 1.0f - x;
sub_result = {0.157138, 0.353809, 0.862595, 0.885693, 0.340074, 0.464184,
0.257084, 0.154395, 0.787718, 0.700299, 0.137829, 0.591059,
0.873153, 0.843381, 0.571159, 0.152347, 0.754137, 0.330954,
0.121117, 0.323741, 0.333547, 0.67477, 0.586061, 0.165859};
check_shape(x.shape, {2, 3, 4});
check_data(reinterpret_cast<const float*>(x.Data()), sub_result.data(),
sub_result.size());
}
TEST(fastdeploy, check_broadcast_dim1) {
@@ -498,6 +507,15 @@ TEST(fastdeploy, mixed_operation) {
check_shape(output.shape, {2, 3, 4});
check_data(reinterpret_cast<const float*>(output.Data()), result.data(),
result.size());
result = {2.854443, 1.87709, 1.585621, 1.012709, 0.332781, 0.998346,
0.228024, 2.140475, 0.246941, 0.301517, 1.575438, 0.595582,
-0.410393, -0.163718, -0.405571, 0.58563, -0.177035, 0.263035,
0.075725, 0.591098, 0.156365, -0.106078, -0.475957, 0.626429};
output = a + b * c / d - e;
check_shape(output.shape, {2, 3, 4});
check_data(reinterpret_cast<const float*>(output.Data()), result.data(),
result.size());
}
} // namespace function