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

1# from .addmm import addmm 

2from .argmax import argmax 

3from .argmin import argmin 

4from .bmm import bmm, bmm_out 

5 

6# from .conv1d import conv1d 

7# from .conv2d import conv2d 

8# from .thnn_conv2d import thnn_conv2d 

9# from .conv_depthwise2d import _conv_depthwise2d 

10# from .flash_attention import flash_attention, scaled_dot_product_attention 

11from .gelu import gelu, gelu_, gelu_backward 

12from .layernorm import layer_norm 

13from .mean import global_avg_pool, mean_dim 

14from .mm import mm, mm_out 

15from .mv import mv 

16from .pow import ( 

17 pow_scalar, 

18 pow_tensor_scalar, 

19 pow_tensor_scalar_, 

20 pow_tensor_tensor, 

21 pow_tensor_tensor_, 

22) 

23from .rsqrt import rsqrt 

24from .sigmoid import sigmoid 

25from .silu import silu 

26from .softmax import softmax 

27from .where import where_scalar_other, where_scalar_self, where_self, where_self_out 

28 

29__all__ = [ 

30 # "addmm", 

31 "argmax", 

32 "argmin", 

33 "bmm", 

34 "bmm_out", 

35 # "conv1d", 

36 # "conv2d", 

37 # "_conv_depthwise2d", 

38 # "flash_attention", 

39 "gelu", 

40 "gelu_", 

41 "gelu_backward", 

42 "global_avg_pool", 

43 "layer_norm", 

44 "mean_dim", 

45 "mm", 

46 "mm_out", 

47 "mv", 

48 "pow_scalar", 

49 "pow_tensor_scalar", 

50 "pow_tensor_scalar_", 

51 "pow_tensor_tensor", 

52 "pow_tensor_tensor_", 

53 "rsqrt", 

54 # "scaled_dot_product_attention", 

55 "sigmoid", 

56 "silu", 

57 "softmax", 

58 # "thnn_conv2d", 

59 "where_scalar_other", 

60 "where_scalar_self", 

61 "where_self", 

62 "where_self_out", 

63]