Coverage for src/flag_gems/runtime/backend/_hygon/ops/__init__.py: 0%

1from .all import all, all_dim, all_dims 

2from .any import any, any_dim, any_dims 

3from .attention import ( 

4 ScaleDotProductAttention, 

5 flash_attention_forward, 

6 flash_attn_varlen_func, 

7 scaled_dot_product_attention, 

8 scaled_dot_product_attention_backward, 

9 scaled_dot_product_attention_forward, 

10) 

11from .div import ( 

12 div_mode, 

13 div_mode_, 

14 floor_divide, 

15 floor_divide_, 

16 remainder, 

17 remainder_, 

18 true_divide, 

19 true_divide_, 

20) 

21from .exponential_ import exponential_ 

22from .fill import fill_scalar, fill_scalar_, fill_tensor, fill_tensor_ 

23from .gelu import gelu, gelu_ 

24from .hadamard_transform import hadamard_transform 

25from .isclose import allclose, isclose 

26from .isin import isin 

27from .matmul_bf16 import matmul_bf16 

28from .matmul_int8 import matmul_int8 

29from .mm import mm 

30from .pow import ( 

31 pow_scalar, 

32 pow_tensor_scalar, 

33 pow_tensor_scalar_, 

34 pow_tensor_tensor, 

35 pow_tensor_tensor_, 

36) 

37from .randperm import randperm 

38from .silu import silu, silu_, silu_backward 

39from .sort import sort, sort_stable 

40from .unique import _unique2 

41from .upsample_nearest2d import upsample_nearest2d 

42 

43__all__ = [ 

44 "_unique2", 

45 "ScaleDotProductAttention", 

46 "all", 

47 "all_dim", 

48 "all_dims", 

49 "allclose", 

50 "any", 

51 "any_dim", 

52 "any_dims", 

53 "div_mode", 

54 "div_mode_", 

55 "exponential_", 

56 "fill_scalar", 

57 "fill_scalar_", 

58 "fill_tensor", 

59 "fill_tensor_", 

60 "flash_attention_forward", 

61 "flash_attn_varlen_func", 

62 "floor_divide", 

63 "floor_divide_", 

64 "gelu", 

65 "gelu_", 

66 "hadamard_transform", 

67 "isin", 

68 "isclose", 

69 "matmul_bf16", 

70 "matmul_int8", 

71 "mm", 

72 "pow_scalar", 

73 "pow_tensor_scalar", 

74 "pow_tensor_scalar_", 

75 "pow_tensor_tensor", 

76 "pow_tensor_tensor_", 

77 "randperm", 

78 "remainder", 

79 "remainder_", 

80 "scaled_dot_product_attention", 

81 "scaled_dot_product_attention_backward", 

82 "scaled_dot_product_attention_forward", 

83 "silu", 

84 "silu_", 

85 "silu_backward", 

86 "sort", 

87 "sort_stable", 

88 "true_divide", 

89 "true_divide_", 

90 "upsample_nearest2d", 

91]