apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-18 14:40:44 +08:00
Code Issues Actions 6 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

21
fastdeploy/function/elementwise.cc
View File

@@ -86,4 +86,25 @@ FDTensor operator/(const FDTensor& x, const FDTensor& y) {
return out;
}
#define INSTANTIATE_OPERATOR(operation_type) \
template FDTensor operator operation_type(const FDTensor& x, bool y); \
template FDTensor operator operation_type(const FDTensor& x, uint8_t y); \
template FDTensor operator operation_type(const FDTensor& x, int16_t y); \
template FDTensor operator operation_type(const FDTensor& x, int y); \
template FDTensor operator operation_type(const FDTensor& x, int64_t y); \
template FDTensor operator operation_type(const FDTensor& x, float y); \
template FDTensor operator operation_type(const FDTensor& x, double y); \
template FDTensor operator operation_type(bool x, const FDTensor& y); \
template FDTensor operator operation_type(uint8_t x, const FDTensor& y); \
template FDTensor operator operation_type(int16_t x, const FDTensor& y); \
template FDTensor operator operation_type(int x, const FDTensor& y); \
template FDTensor operator operation_type(int64_t x, const FDTensor& y); \
template FDTensor operator operation_type(float x, const FDTensor& y); \
template FDTensor operator operation_type(double x, const FDTensor& y)
INSTANTIATE_OPERATOR(+);
INSTANTIATE_OPERATOR(-);
INSTANTIATE_OPERATOR(*);
INSTANTIATE_OPERATOR(/);
} // namespace fastdeploy