|
|
@ -82,6 +82,9 @@ void SparseApplyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) {
|
|
|
|
bool SparseApplyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
|
|
|
|
bool SparseApplyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
|
|
|
|
const std::vector<AddressPtr> &outputs) {
|
|
|
|
const std::vector<AddressPtr> &outputs) {
|
|
|
|
ReInit(inputs);
|
|
|
|
ReInit(inputs);
|
|
|
|
|
|
|
|
if (indices_size_ == 0) {
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
|
|
|
}
|
|
|
|
return Launch(inputs, workspace, outputs);
|
|
|
|
return Launch(inputs, workspace, outputs);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|