fix bug of gpu runtime deteriorate

remove useless ip
This commit is contained in:
mengyuanli 2021-09-17 09:28:08 +08:00
parent 6064b4e1f0
commit 40c678fccb
3 changed files with 2 additions and 7 deletions

View File

@ -826,7 +826,7 @@ std::vector<std::shared_ptr<LiteOpActor>> CreateOpActor(const std::vector<kernel
return actors;
}
int MindrtInit() { return mindspore::Initialize("tcp://127.0.0.1:8080", "", "", ""); }
int MindrtInit() { return mindspore::Initialize("", "", "", ""); }
void MindrtTerminate(const std::vector<std::shared_ptr<LiteOpActor>> &actor_list) {
for (const auto &actor : actor_list) {

View File

@ -329,11 +329,6 @@ void *Tensor::MutableData() {
return this->data_;
}
void *Tensor::data() {
Prepare();
return this->data_;
}
void Tensor::IncRefCount() {
ref_count_++;
if (allocator_ != nullptr) {

View File

@ -119,7 +119,7 @@ class Tensor : public mindspore::tensor::MSTensor {
void *ReallocData();
void *data() override;
void *data() override { return data_; };
virtual void *data() const { return data_; }