Coverage for  / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / platform / mindspore / symmetric_memory / aclshmem_ms / gen_ops_def.py: 0%

1# Copyright 2023 Huawei Technologies Co., Ltd 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================ 

15 

16"""Operators definition generated by gen_ops.py, includes functions.""" 

17 

18from .gen_ops_prim import * 

19 

20 

21def put_mem(target, target_offset, src, src_offset, size, target_pe): 

22 r""" 

23  

24 """ 

25 return put_mem_op(target, target_offset, src, src_offset, size, target_pe) 

26 

27 

28def get_mem(target, target_offset, src, src_offset, size, target_pe=0, non_blocking=False): 

29 r""" 

30  

31 """ 

32 return get_mem_op(target, target_offset, src, src_offset, size, target_pe, non_blocking) 

33 

34 

35def signal_op(signal, signal_offset, signal_value, signal_op=0, target_pe=0): 

36 r""" 

37  

38 """ 

39 return signal_op_op(signal, signal_offset, signal_value, signal_op, target_pe) 

40 

41 

42def put_mem_signal(target, target_offset, src, src_offset, size, signal, signal_offset, signal_value, signal_op=0, target_pe=0): 

43 r""" 

44  

45 """ 

46 return put_mem_signal_op(target, target_offset, src, src_offset, size, signal, signal_offset, signal_value, signal_op, target_pe) 

47 

48 

49def signal_wait_until(depend_target, signal, signal_offset, compare_value, compare_op=0): 

50 r""" 

51  

52 """ 

53 return signal_wait_until_op(depend_target, signal, signal_offset, compare_value, compare_op)