fix memory leak in HogwildWorker (#19956)

fix memory leak in HogwildWorker,  whose ops are  explicitly deleted in destructor
fix-python-transpose
xujiaqi01 5 years ago committed by GitHub
parent b8aff5e5e9
commit f50e701b3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -151,7 +151,12 @@ class CPUWorkerBase : public DeviceWorker {
class HogwildWorker : public CPUWorkerBase { class HogwildWorker : public CPUWorkerBase {
public: public:
HogwildWorker() {} HogwildWorker() {}
virtual ~HogwildWorker() {} virtual ~HogwildWorker() {
for (OperatorBase* op : ops_) {
delete op;
}
std::vector<OperatorBase*>().swap(ops_);
}
virtual void Initialize(const TrainerDesc& desc); virtual void Initialize(const TrainerDesc& desc);
virtual void TrainFiles(); virtual void TrainFiles();
virtual void TrainFilesWithProfiler(); virtual void TrainFilesWithProfiler();

Loading…
Cancel
Save