xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update internal op on git lfs

This commit is contained in:
zhengzuohe 2024-03-29 13:25:04 +08:00
parent 35acddc38e
commit d1d0f85d12
6 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitattributes vendored
View File

@ -1,3 +1,3 @@
mindspore/ccsrc/plugin/device/ascend/kernel/dvm/prebuild/aarch64/libdvm.a filter=lfs diff=lfs merge=lfs -text mindspore/ccsrc/plugin/device/ascend/kernel/dvm/prebuild/aarch64/libdvm.a filter=lfs diff=lfs merge=lfs -text
mindspore/ccsrc/plugin/device/ascend/kernel/dvm/prebuild/x86_64/libdvm.a filter=lfs diff=lfs merge=lfs -text mindspore/ccsrc/plugin/device/ascend/kernel/dvm/prebuild/x86_64/libdvm.a filter=lfs diff=lfs merge=lfs -text
ms_kernels_internal.tar.gz filter=lfs diff=lfs merge=lfs -text mindspore/ccsrc/plugin/device/ascend/kernel/internal/prebuild/ms_kernels_internal.tar.gz filter=lfs diff=lfs merge=lfs -text

12
mindspore/ccsrc/plugin/device/ascend/kernel/internal/OWNERS Normal file
View File

@ -0,0 +1,12 @@
approvers:
- zhaizhiqiang #
- gaoxiong1
- ckey_dou
- anyrenwei
- zhangxuetong
- liulili-huawei
- zichun_ye
- dayschan
options:
no_parent_owners: true

5
mindspore/ccsrc/plugin/device/ascend/kernel/internal/internal_kernel_mod.cc
View File

@ -48,7 +48,10 @@ int InternalKernelMod::Build(const std::vector<KernelTensor *> &inputs, const st
info.output_dtype_.emplace_back(InternalKernelUtils::ToInternalDType(outputs[iter->first]->dtype_id())); info.output_dtype_.emplace_back(InternalKernelUtils::ToInternalDType(outputs[iter->first]->dtype_id()));
info.output_format_.emplace_back(InternalKernelUtils::ToInternalFormat(outputs[iter->first]->format())); info.output_format_.emplace_back(InternalKernelUtils::ToInternalFormat(outputs[iter->first]->format()));
} }
impl_->Init(info); if (!impl_->Init(info)) {
MS_LOG(ERROR) << "Internal Op '" << kernel_name_ << "' is initialized FAILED.";
return 1;
}
for (auto iter = inputsIdxMap_.begin(); iter != inputsIdxMap_.end(); iter++) { for (auto iter = inputsIdxMap_.begin(); iter != inputsIdxMap_.end(); iter++) {
InternalKernelUtils::ToInternalTensor(inputs_[iter->second], inputs[iter->first]); InternalKernelUtils::ToInternalTensor(inputs_[iter->second], inputs[iter->first]);
} }

3
mindspore/ccsrc/plugin/device/ascend/kernel/internal/prebuild/ms_kernels_internal.tar.gz Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4997c83144ba095c5ef85a55274a102789f2795c903135ac36e3a0f61aa3a388
size 34965262

3
ms_kernels_internal.tar.gz
View File

@ -1,3 +0,0 @@
version https://git-lfs.github.com/spec/v1
oid sha256:322f1385a3cce823f07501df995cf8f0231f4e56191d29dce4c5a1eb4192e0da
size 34548261

7
scripts/build/check_and_build_ms_kernels_internal.sh
View File

@ -18,14 +18,15 @@ if [[ "$(uname)" == Linux && "$(arch)" == aarch64 ]]; then
if [ -n "${MS_INTERNAL_KERNEL_HOME}" ]; then if [ -n "${MS_INTERNAL_KERNEL_HOME}" ]; then
echo "Use local MS_INTERNAL_KERNEL_HOME : ${MS_INTERNAL_KERNEL_HOME}" echo "Use local MS_INTERNAL_KERNEL_HOME : ${MS_INTERNAL_KERNEL_HOME}"
else else
lib_file=${BASEPATH}/ms_kernels_internal.tar.gz file_path=${BASEPATH}/mindspore/ccsrc/plugin/device/ascend/kernel/internal/prebuild
lib_file=${file_path}/ms_kernels_internal.tar.gz
if [ -f "${lib_file}" ]; then if [ -f "${lib_file}" ]; then
file_lines=`cat "${lib_file}" | wc -l` file_lines=`cat "${lib_file}" | wc -l`
if [ ${file_lines} -ne 3 ]; then if [ ${file_lines} -ne 3 ]; then
tar -zxf ${lib_file} -C ${BASEPATH} tar -zxf ${lib_file} -C ${file_path}
if [ $? -eq 0 ]; then if [ $? -eq 0 ]; then
echo "Unzip ms_kernel_internal.tar.gz SUCCESS!" echo "Unzip ms_kernel_internal.tar.gz SUCCESS!"
export MS_INTERNAL_KERNEL_HOME="${BASEPATH}/ms_kernels_internal" export MS_INTERNAL_KERNEL_HOME="${file_path}/ms_kernels_internal"
echo "MS_INTERNAL_KERNEL_HOME = ${MS_INTERNAL_KERNEL_HOME}" echo "MS_INTERNAL_KERNEL_HOME = ${MS_INTERNAL_KERNEL_HOME}"
else else
echo "[WARNING] Unzip ms_kernel_internal.tar.gz FAILED!" echo "[WARNING] Unzip ms_kernel_internal.tar.gz FAILED!"