mindspore 2.2.14__cp39-cp39-win_amd64.whl → 2.3.0__cp39-cp39-win_amd64.whl

mindspore/ops/_op_impl/tbe/mul_ds.py

@@ -1,38 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Mul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-mul_ds_op_info = TBERegOp("Mul") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("mul.so") \
-    .compute_cost(10) \
-    .kernel_name("mul") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "y", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(mul_ds_op_info)
-def _mul_ds_tbe():
-    """Mul TBE register"""
-    return

mindspore/ops/_op_impl/tbe/mul_no_nan.py

@@ -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.
-# ============================================================================
-
-"""MulNoNan op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-mul_no_nan_op_info = TBERegOp("MulNoNan") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("mul_no_nan.so") \
-    .compute_cost(10) \
-    .kernel_name("mul_no_nan") \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(mul_no_nan_op_info)
-def _mul_no_nan_tbe():
-    """MulNoNan TBE register"""
-    return

mindspore/ops/_op_impl/tbe/mul_no_nan_ds.py

@@ -1,40 +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.
-# ============================================================================
-
-"""MulNoNan op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-mul_no_nan_op_info = TBERegOp("MulNoNan") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("mul_no_nan.so") \
-    .compute_cost(10) \
-    .kernel_name("mul_no_nan") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(mul_no_nan_op_info)
-def _mul_no_nan_ds_tbe():
-    """MulNoNan TBE register"""
-    return

mindspore/ops/_op_impl/tbe/multilabel_margin_loss.py

@@ -1,39 +0,0 @@
-# Copyright 2022 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.
-# ============================================================================
-
-"""MultilabelMarginLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-multilabel_margin_loss_op_info = TBERegOp("MultilabelMarginLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("multilabel_margin_loss.so") \
-    .compute_cost(10) \
-    .kernel_name("multilabel_margin_loss") \
-    .partial_flag(True) \
-    .attr("reduction", "optional", "str", "mean,none,sum", "mean") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "target", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .output(1, "is_target", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(multilabel_margin_loss_op_info)
-def _multilabel_margin_loss_tbe():
-    """MultilabelMarginLoss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/neg.py

@@ -1,39 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Neg op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-neg_op_info = TBERegOp("Neg") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("neg.so") \
-    .compute_cost(10) \
-    .kernel_name("neg") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I8_None, DataType.I8_None) \
-    .get_op_info()
-
-
-@op_info_register(neg_op_info)
-def _neg_tbe():
-    """Neg TBE register"""
-    return

mindspore/ops/_op_impl/tbe/neg_ds.py

@@ -1,40 +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.
-# ============================================================================
-
-"""Neg op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-neg_ds_op_info = TBERegOp("Neg") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("neg.so") \
-    .compute_cost(10) \
-    .kernel_name("neg") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I8_None, DataType.I8_None) \
-    .get_op_info()
-
-
-@op_info_register(neg_ds_op_info)
-def _neg_ds_tbe():
-    """Neg TBE register"""
-    return

mindspore/ops/_op_impl/tbe/new_im2col.py

@@ -1,40 +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.
-# ============================================================================
-
-"""new im2col op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-new_im2col_op_info = TBERegOp("NewIm2Col") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("im2col.so") \
-    .compute_cost(10) \
-    .kernel_name("im2col") \
-    .attr("ksizes", "required", "listInt", "all") \
-    .attr("strides", "optional", "listInt", "all", "1") \
-    .attr("dilations", "optional", "listInt", "all", "1") \
-    .attr("padding_mode", "optional", "str", "all", "SAME") \
-    .attr("pad_list", "optional", "listInt", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_Default) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_Default) \
-    .get_op_info()
-
-
-@op_info_register(new_im2col_op_info)
-def _new_im2col_tbe():
-    """NewIm2Col TBE register"""
-    return

mindspore/ops/_op_impl/tbe/nll_loss.py

@@ -1,41 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NLLLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-nll_loss_op_info = TBERegOp("NLLLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("nll_loss.so") \
-    .compute_cost(10) \
-    .kernel_name("nll_loss") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "target", False, "required", "all") \
-    .input(2, "weight", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .output(1, "total_weight", False, "optional", "all") \
-    .attr("reduction", "optional", "str", "all", "mean") \
-    .attr("ignore_index", "optional", "int", "all", "-100") \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(nll_loss_op_info)
-def _nll_loss_tbe():
-    """NLLLoss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/nll_loss_grad.py

@@ -1,44 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NLLLossGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-nll_loss_grad_op_info = TBERegOp("NLLLossGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("nll_loss_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("nll_loss_grad") \
-    .partial_flag(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "y_grad", False, "required", "all") \
-    .input(2, "target", False, "required", "all") \
-    .input(3, "weight", False, "required", "all") \
-    .input(4, "total_weight", False, "required", "all") \
-    .output(0, "x_grad", False, "required", "all") \
-    .attr("reduction", "optional", "str", "all", "mean") \
-    .attr("ignore_index", "optional", "int", "all", "-100") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(nll_loss_grad_op_info)
-def _nll_loss_grad_tbe():
-    """NLLLossGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/nms_with_mask.py

@@ -1,39 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NMSWithMask op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-nms_with_mask_op_info = TBERegOp("NMSWithMask") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("nms_with_mask.so") \
-    .compute_cost(10) \
-    .kernel_name("nms_with_mask") \
-    .partial_flag(True) \
-    .attr("iou_threshold", "optional", "float", "all") \
-    .input(0, "box_scores", False, "required", "all") \
-    .output(0, "selected_boxes", False, "required", "all") \
-    .output(0, "selected_idx", False, "required", "all") \
-    .output(0, "selected_mask", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, DataType.U8_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, DataType.BOOL_Default) \
-    .get_op_info()
-
-
-@op_info_register(nms_with_mask_op_info)
-def _nms_with_mask_tbe():
-    """NMSWithMask TBE register"""
-    return

mindspore/ops/_op_impl/tbe/not_equal.py

@@ -1,41 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NotEqual op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-not_equal_op_info = TBERegOp("NotEqual") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("not_equal.so") \
-    .compute_cost(10) \
-    .kernel_name("not_equal") \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.I8_None, DataType.I8_None, DataType.BOOL_None) \
-    .dtype_format(DataType.U8_None, DataType.U8_None, DataType.BOOL_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.BOOL_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.BOOL_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(not_equal_op_info)
-def _not_equal_tbe():
-    """Equal TBE register"""
-    return

mindspore/ops/_op_impl/tbe/not_equal_ds.py

@@ -1,42 +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.
-# ============================================================================
-
-"""NotEqual op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-not_equal_ds_op_info = TBERegOp("NotEqual") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("not_equal.so") \
-    .compute_cost(10) \
-    .kernel_name("not_equal") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.I8_None, DataType.I8_None, DataType.BOOL_None) \
-    .dtype_format(DataType.U8_None, DataType.U8_None, DataType.BOOL_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.BOOL_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.BOOL_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(not_equal_ds_op_info)
-def _not_ds_equal_tbe():
-    """Equal TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_alloc_float_status.py

@@ -1,34 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NPUAllocFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_alloc_float_status_op_info = TBERegOp("NPUAllocFloatStatus") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_alloc_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_alloc_float_status") \
-    .partial_flag(True) \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_alloc_float_status_op_info)
-def _npu_alloc_float_status_tbe():
-    """NPUAllocFloatStatus TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_clear_float_status.py

@@ -1,35 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NPUClearFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_clear_float_status_op_info = TBERegOp("NPUClearFloatStatus") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_clear_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_clear_float_status") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_clear_float_status_op_info)
-def _npu_clear_float_status_tbe():
-    """NPUClearFloatStatus TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py

@@ -1,35 +0,0 @@
-# Copyright 2023 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.
-# ============================================================================
-
-"""NPUClearFloatStatusV2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_clear_float_status_v2_op_info = TBERegOp("NPUClearFloatStatusV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_clear_float_status_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_clear_float_status_v2") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_clear_float_status_v2_op_info)
-def _npu_clear_float_status_v2_tbe():
-    """NPUClearFloatStatusV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_get_float_status.py

@@ -1,35 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""NPUGetFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_get_float_status_op_info = TBERegOp("NPUGetFloatStatus") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_get_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_get_float_status") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_get_float_status_op_info)
-def _npu_get_float_status_tbe():
-    """NPUGetFloatStatus TBE register"""
-    return