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

1from .argmax import argmax 

2from .attention import ( 

3 ScaleDotProductAttention, 

4 flash_attention_forward, 

5 flash_attn_varlen_func, 

6 scaled_dot_product_attention, 

7 scaled_dot_product_attention_backward, 

8 scaled_dot_product_attention_forward, 

9) 

10from .cat import cat 

11from .count_nonzero import count_nonzero 

12from .flash_api import mha_fwd, mha_varlan_fwd 

13from .hstack import hstack 

14from .index import index 

15from .isin import isin 

16from .kron import kron 

17from .masked_select import masked_select 

18from .matmul_bf16 import matmul_bf16 

19from .matmul_int8 import matmul_int8 

20from .mm import mm, mm_out 

21from .normal import ( 

22 normal_, 

23 normal_distribution, 

24 normal_float_tensor, 

25 normal_tensor_float, 

26 normal_tensor_tensor, 

27) 

28from .pow import ( 

29 pow_scalar, 

30 pow_tensor_scalar, 

31 pow_tensor_scalar_, 

32 pow_tensor_tensor, 

33 pow_tensor_tensor_, 

34) 

35from .randn import randn 

36from .randn_like import randn_like 

37from .rms_norm import rms_norm 

38from .rsqrt import rsqrt, rsqrt_ 

39from .silu_and_mul import silu_and_mul, silu_and_mul_out 

40from .stack import stack 

41from .unique import _unique2 

42from .vdot import vdot 

43from .zeros import zero_, zeros 

44from .zeros_like import zeros_like 

45 

46__all__ = [ 

47 "argmax", 

48 "cat", 

49 "count_nonzero", 

50 "hstack", 

51 "isin", 

52 "kron", 

53 "matmul_bf16", 

54 "matmul_int8", 

55 "masked_select", 

56 "mm", 

57 "mm_out", 

58 "randn", 

59 "randn_like", 

60 "rms_norm", 

61 "stack", 

62 "kron", 

63 "isin", 

64 "_unique2", 

65 "zeros", 

66 "zeros_like", 

67 "zero_", 

68 "pow_scalar", 

69 "pow_tensor_scalar", 

70 "pow_tensor_scalar_", 

71 "pow_tensor_tensor", 

72 "pow_tensor_tensor_", 

73 "rsqrt", 

74 "rsqrt_", 

75 "silu_and_mul", 

76 "silu_and_mul_out", 

77 "ScaleDotProductAttention", 

78 "flash_attention_forward", 

79 "flash_attn_varlen_func", 

80 "scaled_dot_product_attention", 

81 "scaled_dot_product_attention_backward", 

82 "scaled_dot_product_attention_forward", 

83 "mha_fwd", 

84 "mha_varlan_fwd", 

85 "vdot", 

86 "index", 

87 "normal_", 

88 "normal_distribution", 

89 "normal_float_tensor", 

90 "normal_tensor_float", 

91 "normal_tensor_tensor", 

92]