回退 'Pull Request !11074 : replace tdt with acltdt interface'
gongxiaoqing
4 years ago
committed by
Gitee
parent
26d4b99019
commit
7f538b51e7
@ -1,6 +1,5 @@
|
||||
file(
|
||||
GLOB_RECURSE _CURRENT_SRC_FILES
|
||||
RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"*.cc")
|
||||
file(GLOB_RECURSE _CURRENT_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
set_property(SOURCE ${_CURRENT_SRC_FILES} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_MD)
|
||||
add_library(engine-tdt OBJECT tdt_plugin.cc tdt_handle.cc)
|
||||
add_library(engine-tdt OBJECT
|
||||
tdt_plugin.cc
|
||||
)
|
||||
|
||||
@ -1,39 +0,0 @@
|
||||
/**
|
||||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "minddata/dataset/engine/tdt/tdt_handle.h"
|
||||
namespace mindspore {
|
||||
namespace dataset {
|
||||
|
||||
std::vector<acltdtChannelHandle *> TdtHandle::acl_handle = std::vector<acltdtChannelHandle *>();
|
||||
|
||||
void TdtHandle::AddHandle(acltdtChannelHandle *handle) {
|
||||
if (handle != nullptr) {
|
||||
acl_handle.emplace_back(handle);
|
||||
}
|
||||
}
|
||||
|
||||
bool TdtHandle::DestroyHandle() {
|
||||
for (auto handle : acl_handle) {
|
||||
if (handle != nullptr) {
|
||||
if (acltdtDestroyChannel(handle) != ACL_SUCCESS) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
} // namespace dataset
|
||||
} // namespace mindspore
|
||||
@ -1,38 +0,0 @@
|
||||
/**
|
||||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_MINDDATA_DATASET_ENGINE_TDT_TDT_HANDLE_H_
|
||||
#define MINDSPORE_CCSRC_MINDDATA_DATASET_ENGINE_TDT_TDT_HANDLE_H_
|
||||
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include "acl/acl_tdt.h"
|
||||
|
||||
namespace mindspore {
|
||||
namespace dataset {
|
||||
class TdtHandle {
|
||||
public:
|
||||
static void AddHandle(acltdtChannelHandle *handle);
|
||||
|
||||
static bool DestroyHandle();
|
||||
|
||||
private:
|
||||
TdtHandle() {}
|
||||
|
||||
static std::vector<acltdtChannelHandle *> acl_handle;
|
||||
};
|
||||
} // namespace dataset
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_MINDDATA_DATASET_ENGINE_TDT_TDT_HANDLE_H_
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in new issue