Skip to content

Commit

Permalink
Merge branch 'main' into hanjr-patch
Browse files Browse the repository at this point in the history
  • Loading branch information
luomai committed May 10, 2023
2 parents 01cfd3e + 004017b commit aed6e99
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions chapter_backend_and_runtime/compute_schedule_and_execute.md
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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)
Expand All @@ -105,15 +101,15 @@ 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")
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)
Expand Down

0 comments on commit aed6e99

Please sign in to comment.