From 004017beaaa125acd67e8c9d455b12e143bb1266 Mon Sep 17 00:00:00 2001
From: jojobugfree <68210792+jojobugfree@users.noreply.github.com>
Date: Wed, 10 May 2023 14:16:14 +0800
Subject: [PATCH] Adapt the mindspore code used in the book to the MS2.0
 version (#455)

Co-authored-by: caifubi <caifubi1@huawei.com>
---
 .../compute_schedule_and_execute.md                  | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/chapter_backend_and_runtime/compute_schedule_and_execute.md b/chapter_backend_and_runtime/compute_schedule_and_execute.md
index 13133d7..bfa1131 100644
--- a/chapter_backend_and_runtime/compute_schedule_and_execute.md
+++ b/chapter_backend_and_runtime/compute_schedule_and_execute.md
@@ -9,10 +9,6 @@
 ```python
 import mindspore.nn as nn
 from mindspore import context
-from mindspore import ms_function
-
-# 以单算子方式执行后续计算中的算子。
-context.set_context(mode=context.PYNATIVE_MODE)
 
 class Computation(nn.Cell):
    def construct(self, x, y):
@@ -71,14 +67,14 @@ Kernel。主流框架均提供了指定算子所在运行设备的能力，以Mi
 import numpy as np
 from mindspore import Tensor
 import mindspore.ops.operations as ops
-from mindspore.common.api import ms_function
+from mindspore.common.api import jit
 
 # 创建算子并指定执行算子的硬件设备
 add = ops.Add().add_prim_attr('primitive_target', 'CPU')
 sub = ops.Sub().add_prim_attr('primitive_target', 'GPU')
 
 # 指定按照静态计算图模式执行函数
-@ms_function
+@jit
 def compute(x, y, z):
     r = add(x, y)
     return sub(r, z)
@@ -105,7 +101,7 @@ output = compute(x, y, z)
 import mindspore as ms
 from mindspore import Parameter, Tensor
 import mindspore.ops.operations as ops
-from mindspore.common.api import ms_function
+from mindspore.common.api import jit
 
 # 定义全局变量
 x = Parameter(Tensor([1.0], ms.float32), name="x")
@@ -113,7 +109,7 @@ y = Tensor([0.2], ms.float32)
 z = Tensor([0.3], ms.float32)
 
 # 指定按照静态计算图模式执行函数
-@ms_function
+@jit
 def compute(y, z):
     ops.Assign()(x, y)
     ops.Assign()(x, z)