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 

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 

12from .layernorm import layer_norm 

13from .mean import global_avg_pool, mean_dim 

14from .mm import mm 

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 # "conv1d", 

35 # "conv2d", 

36 # "_conv_depthwise2d", 

37 # "flash_attention", 

38 "gelu", 

39 "global_avg_pool", 

40 "layer_norm", 

41 "mean_dim", 

42 "mm", 

43 "mv", 

44 "pow_scalar", 

45 "pow_tensor_scalar", 

46 "pow_tensor_scalar_", 

47 "pow_tensor_tensor", 

48 "pow_tensor_tensor_", 

49 "rsqrt", 

50 # "scaled_dot_product_attention", 

51 "sigmoid", 

52 "silu", 

53 "softmax", 

54 # "thnn_conv2d", 

55 "where_scalar_other", 

56 "where_scalar_self", 

57 "where_self", 

58 "where_self_out", 

59]