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

1from ._safe_softmax import _safe_softmax 

2from ._upsample_nearest_exact1d import _upsample_nearest_exact1d 

3from .abs import abs, abs_ 

4from .add import add, add_ 

5from .addmm import addmm, addmm_out 

6from .angle import angle 

7from .arcsinh import arcsinh, arcsinh_out 

8from .attention import ( 

9 ScaleDotProductAttention, 

10 flash_attention_forward, 

11 flash_attn_varlen_func, 

12 scaled_dot_product_attention, 

13 scaled_dot_product_attention_backward, 

14 scaled_dot_product_attention_forward, 

15) 

16from .bitwise_and import ( 

17 bitwise_and_scalar, 

18 bitwise_and_scalar_, 

19 bitwise_and_scalar_tensor, 

20 bitwise_and_tensor, 

21 bitwise_and_tensor_, 

22) 

23from .bitwise_left_shift import ( 

24 bitwise_left_shift, 

25 bitwise_left_shift_, 

26 bitwise_left_shift_out, 

27) 

28from .bitwise_right_shift import ( 

29 bitwise_right_shift, 

30 bitwise_right_shift_, 

31 bitwise_right_shift_out, 

32) 

33from .clamp import clamp, clamp_, clamp_min, clamp_min_, clamp_tensor, clamp_tensor_ 

34from .conv2d import conv2d 

35from .cos import cos, cos_ 

36from .count_nonzero import count_nonzero 

37from .cumsum import cumsum, cumsum_out, normed_cumsum 

38from .div import ( 

39 div_mode, 

40 div_mode_, 

41 floor_divide, 

42 floor_divide_, 

43 remainder, 

44 remainder_, 

45 true_divide, 

46 true_divide_, 

47 true_divide_out, 

48) 

49from .dropout import dropout, dropout_backward 

50from .embedding import embedding, embedding_backward 

51from .eq import eq, eq_scalar, equal 

52from .exponential_ import exponential_ 

53from .fill import ( 

54 fill_scalar, 

55 fill_scalar_, 

56 fill_scalar_out, 

57 fill_tensor, 

58 fill_tensor_, 

59 fill_tensor_out, 

60) 

61from .gather import gather, gather_backward 

62from .ge import ge, ge_scalar 

63from .gelu import gelu, gelu_, gelu_backward 

64from .i0 import i0, i0_out 

65from .i0_ import i0_ 

66from .index_add import index_add, index_add_ 

67from .index_put import index_put, index_put_ 

68from .index_select import index_select 

69from .isin import isin 

70from .isnan import isnan 

71from .layernorm import layer_norm, layer_norm_backward 

72from .lift_fresh_copy import lift_fresh_copy, lift_fresh_copy_out 

73from .linspace import linspace 

74from .log_softmax import log_softmax, log_softmax_backward 

75from .logaddexp import logaddexp, logaddexp_out 

76from .logical_and import logical_and 

77from .logical_or import logical_or, logical_or_ 

78from .margin_ranking_loss import margin_ranking_loss 

79from .masked_select import masked_select 

80from .mean import mean, mean_dim 

81from .mul import mul, mul_ 

82from .mv import mv 

83from .neg import neg, neg_ 

84from .nonzero import nonzero 

85from .pad import constant_pad_nd, pad 

86from .polar import polar 

87from .pow import ( 

88 pow_scalar, 

89 pow_tensor_scalar, 

90 pow_tensor_scalar_, 

91 pow_tensor_tensor, 

92 pow_tensor_tensor_, 

93) 

94from .prelu import prelu 

95from .quantile import quantile 

96from .randperm import randperm 

97from .reflection_pad2d import reflection_pad2d 

98from .repeat import repeat 

99from .repeat_interleave import ( 

100 repeat_interleave_self_int, 

101 repeat_interleave_self_tensor, 

102 repeat_interleave_tensor, 

103) 

104from .rms_norm import rms_norm, rms_norm_backward, rms_norm_forward 

105from .scatter import scatter, scatter_ 

106from .select_backward import select_backward 

107from .sigmoid import sigmoid, sigmoid_, sigmoid_backward 

108from .soft_margin_loss import soft_margin_loss, soft_margin_loss_out 

109from .softmax import softmax, softmax_backward 

110from .sort import sort, sort_stable 

111from .special_i0e import special_i0e, special_i0e_out 

112from .special_i1 import special_i1, special_i1_out 

113from .sum import sum, sum_dim, sum_dim_out, sum_out 

114from .t_copy import t_copy, t_copy_out 

115from .tile import tile 

116from .to import to_copy 

117from .topk import topk 

118from .triu import triu 

119from .unique import _unique2 

120from .upsample_bicubic2d import upsample_bicubic2d 

121from .upsample_linear1d import upsample_linear1d 

122from .upsample_nearest2d import upsample_nearest2d 

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

124from .zero import zero, zero_out 

125 

126__all__ = [ 

127 "_safe_softmax", 

128 "_upsample_nearest_exact1d", 

129 "abs", 

130 "abs_", 

131 "add", 

132 "add_", 

133 "addmm", 

134 "addmm_out", 

135 "angle", 

136 "arcsinh", 

137 "arcsinh_out", 

138 "bitwise_and_scalar", 

139 "bitwise_and_scalar_", 

140 "bitwise_and_scalar_tensor", 

141 "bitwise_and_tensor", 

142 "bitwise_and_tensor_", 

143 "bitwise_left_shift", 

144 "bitwise_left_shift_", 

145 "bitwise_left_shift_out", 

146 "bitwise_right_shift", 

147 "bitwise_right_shift_", 

148 "bitwise_right_shift_out", 

149 "clamp", 

150 "clamp_", 

151 "clamp_tensor", 

152 "clamp_tensor_", 

153 "clamp_min", 

154 "clamp_min_", 

155 "conv2d", 

156 "cos", 

157 "cos_", 

158 "count_nonzero", 

159 "cumsum", 

160 "cumsum_out", 

161 "normed_cumsum", 

162 "div_mode", 

163 "div_mode_", 

164 "embedding", 

165 "embedding_backward", 

166 "floor_divide", 

167 "floor_divide_", 

168 "remainder", 

169 "remainder_", 

170 "true_divide", 

171 "true_divide_", 

172 "true_divide_out", 

173 "dropout", 

174 "dropout_backward", 

175 "eq", 

176 "eq_scalar", 

177 "equal", 

178 "exponential_", 

179 "fill_scalar", 

180 "fill_scalar_", 

181 "fill_scalar_out", 

182 "fill_tensor", 

183 "fill_tensor_", 

184 "fill_tensor_out", 

185 "flash_attention_forward", 

186 "flash_attn_varlen_func", 

187 "gather", 

188 "gather_backward", 

189 "ge", 

190 "ge_scalar", 

191 "gelu", 

192 "gelu_", 

193 "gelu_backward", 

194 "i0", 

195 "i0_out", 

196 "i0_", 

197 "index_add", 

198 "index_add_", 

199 "index_put", 

200 "index_put_", 

201 "index_select", 

202 "isin", 

203 "isnan", 

204 "layer_norm", 

205 "layer_norm_backward", 

206 "lift_fresh_copy", 

207 "lift_fresh_copy_out", 

208 "linspace", 

209 "log_softmax", 

210 "log_softmax_backward", 

211 "logaddexp", 

212 "logaddexp_out", 

213 "logical_and", 

214 "logical_or", 

215 "logical_or_", 

216 "margin_ranking_loss", 

217 "masked_select", 

218 "mean", 

219 "mean_dim", 

220 "mul", 

221 "mul_", 

222 "mv", 

223 "neg", 

224 "neg_", 

225 "nonzero", 

226 "pad", 

227 "polar", 

228 "constant_pad_nd", 

229 "pow_scalar", 

230 "pow_tensor_scalar", 

231 "pow_tensor_scalar_", 

232 "pow_tensor_tensor", 

233 "pow_tensor_tensor_", 

234 "prelu", 

235 "quantile", 

236 "randperm", 

237 "reflection_pad2d", 

238 "repeat", 

239 "repeat_interleave_self_int", 

240 "repeat_interleave_self_tensor", 

241 "repeat_interleave_tensor", 

242 "rms_norm", 

243 "rms_norm_forward", 

244 "rms_norm_backward", 

245 "scaled_dot_product_attention", 

246 "scaled_dot_product_attention_backward", 

247 "scaled_dot_product_attention_forward", 

248 "scatter", 

249 "scatter_", 

250 "select_backward", 

251 "sigmoid", 

252 "sigmoid_", 

253 "sigmoid_backward", 

254 "soft_margin_loss", 

255 "soft_margin_loss_out", 

256 "softmax", 

257 "softmax_backward", 

258 "sort", 

259 "sort_stable", 

260 "special_i0e", 

261 "special_i0e_out", 

262 "special_i1", 

263 "special_i1_out", 

264 "sum", 

265 "sum_dim", 

266 "sum_dim_out", 

267 "sum_out", 

268 "t_copy", 

269 "t_copy_out", 

270 "ScaleDotProductAttention", 

271 "tile", 

272 "to_copy", 

273 "topk", 

274 "triu", 

275 "_unique2", 

276 "upsample_bicubic2d", 

277 "upsample_linear1d", 

278 "upsample_nearest2d", 

279 "where_scalar_other", 

280 "where_scalar_self", 

281 "where_self", 

282 "where_self_out", 

283 "zero", 

284 "zero_out", 

285]