1 // @generated by torchgen/gen.py from
2 // external_functions_codegen_template.cpp
3 #include <torch/csrc/jit/tensorexpr/external_functions.h>
4
5 #include <ATen/Functions.h>
6 #include <ATen/NativeFunctions.h>
7 #include <torch/csrc/jit/tensorexpr/external_functions_registry.h>
8
9 namespace torch::jit::tensorexpr {
10
11 #ifdef C10_MOBILE
12 extern "C" {
13 #endif
14
15 #ifndef C10_MOBILE
nnc_aten_abs(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)16 static void nnc_aten_abs(
17 int64_t bufs_num,
18 void** buf_data,
19 int64_t* buf_ranks,
20 int64_t* buf_dims,
21 int64_t* buf_strides,
22 int8_t* buf_dtypes,
23 int64_t args_num,
24 int64_t* extra_args) {
25 std::vector<at::Tensor> tensors = constructTensors(
26 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
27 at::Tensor& r = tensors[0];
28 const at::Tensor& self = tensors[1];
29 try {
30 at::abs_out(r, self);
31 } catch (...) {
32 }
33 }
nnc_aten_absolute(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)34 static void nnc_aten_absolute(
35 int64_t bufs_num,
36 void** buf_data,
37 int64_t* buf_ranks,
38 int64_t* buf_dims,
39 int64_t* buf_strides,
40 int8_t* buf_dtypes,
41 int64_t args_num,
42 int64_t* extra_args) {
43 std::vector<at::Tensor> tensors = constructTensors(
44 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
45 at::Tensor& r = tensors[0];
46 const at::Tensor& self = tensors[1];
47 try {
48 at::absolute_out(r, self);
49 } catch (...) {
50 }
51 }
nnc_aten_angle(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)52 static void nnc_aten_angle(
53 int64_t bufs_num,
54 void** buf_data,
55 int64_t* buf_ranks,
56 int64_t* buf_dims,
57 int64_t* buf_strides,
58 int8_t* buf_dtypes,
59 int64_t args_num,
60 int64_t* extra_args) {
61 std::vector<at::Tensor> tensors = constructTensors(
62 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
63 at::Tensor& r = tensors[0];
64 const at::Tensor& self = tensors[1];
65 try {
66 at::angle_out(r, self);
67 } catch (...) {
68 }
69 }
nnc_aten_sgn(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)70 static void nnc_aten_sgn(
71 int64_t bufs_num,
72 void** buf_data,
73 int64_t* buf_ranks,
74 int64_t* buf_dims,
75 int64_t* buf_strides,
76 int8_t* buf_dtypes,
77 int64_t args_num,
78 int64_t* extra_args) {
79 std::vector<at::Tensor> tensors = constructTensors(
80 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
81 at::Tensor& r = tensors[0];
82 const at::Tensor& self = tensors[1];
83 try {
84 at::sgn_out(r, self);
85 } catch (...) {
86 }
87 }
nnc_aten_acos(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)88 static void nnc_aten_acos(
89 int64_t bufs_num,
90 void** buf_data,
91 int64_t* buf_ranks,
92 int64_t* buf_dims,
93 int64_t* buf_strides,
94 int8_t* buf_dtypes,
95 int64_t args_num,
96 int64_t* extra_args) {
97 std::vector<at::Tensor> tensors = constructTensors(
98 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
99 at::Tensor& r = tensors[0];
100 const at::Tensor& self = tensors[1];
101 try {
102 at::acos_out(r, self);
103 } catch (...) {
104 }
105 }
nnc_aten_arccos(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)106 static void nnc_aten_arccos(
107 int64_t bufs_num,
108 void** buf_data,
109 int64_t* buf_ranks,
110 int64_t* buf_dims,
111 int64_t* buf_strides,
112 int8_t* buf_dtypes,
113 int64_t args_num,
114 int64_t* extra_args) {
115 std::vector<at::Tensor> tensors = constructTensors(
116 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
117 at::Tensor& r = tensors[0];
118 const at::Tensor& self = tensors[1];
119 try {
120 at::arccos_out(r, self);
121 } catch (...) {
122 }
123 }
nnc_aten_acosh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)124 static void nnc_aten_acosh(
125 int64_t bufs_num,
126 void** buf_data,
127 int64_t* buf_ranks,
128 int64_t* buf_dims,
129 int64_t* buf_strides,
130 int8_t* buf_dtypes,
131 int64_t args_num,
132 int64_t* extra_args) {
133 std::vector<at::Tensor> tensors = constructTensors(
134 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
135 at::Tensor& r = tensors[0];
136 const at::Tensor& self = tensors[1];
137 try {
138 at::acosh_out(r, self);
139 } catch (...) {
140 }
141 }
nnc_aten_arccosh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)142 static void nnc_aten_arccosh(
143 int64_t bufs_num,
144 void** buf_data,
145 int64_t* buf_ranks,
146 int64_t* buf_dims,
147 int64_t* buf_strides,
148 int8_t* buf_dtypes,
149 int64_t args_num,
150 int64_t* extra_args) {
151 std::vector<at::Tensor> tensors = constructTensors(
152 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
153 at::Tensor& r = tensors[0];
154 const at::Tensor& self = tensors[1];
155 try {
156 at::arccosh_out(r, self);
157 } catch (...) {
158 }
159 }
nnc_aten_asinh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)160 static void nnc_aten_asinh(
161 int64_t bufs_num,
162 void** buf_data,
163 int64_t* buf_ranks,
164 int64_t* buf_dims,
165 int64_t* buf_strides,
166 int8_t* buf_dtypes,
167 int64_t args_num,
168 int64_t* extra_args) {
169 std::vector<at::Tensor> tensors = constructTensors(
170 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
171 at::Tensor& r = tensors[0];
172 const at::Tensor& self = tensors[1];
173 try {
174 at::asinh_out(r, self);
175 } catch (...) {
176 }
177 }
nnc_aten_arcsinh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)178 static void nnc_aten_arcsinh(
179 int64_t bufs_num,
180 void** buf_data,
181 int64_t* buf_ranks,
182 int64_t* buf_dims,
183 int64_t* buf_strides,
184 int8_t* buf_dtypes,
185 int64_t args_num,
186 int64_t* extra_args) {
187 std::vector<at::Tensor> tensors = constructTensors(
188 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
189 at::Tensor& r = tensors[0];
190 const at::Tensor& self = tensors[1];
191 try {
192 at::arcsinh_out(r, self);
193 } catch (...) {
194 }
195 }
nnc_aten_atanh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)196 static void nnc_aten_atanh(
197 int64_t bufs_num,
198 void** buf_data,
199 int64_t* buf_ranks,
200 int64_t* buf_dims,
201 int64_t* buf_strides,
202 int8_t* buf_dtypes,
203 int64_t args_num,
204 int64_t* extra_args) {
205 std::vector<at::Tensor> tensors = constructTensors(
206 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
207 at::Tensor& r = tensors[0];
208 const at::Tensor& self = tensors[1];
209 try {
210 at::atanh_out(r, self);
211 } catch (...) {
212 }
213 }
nnc_aten_arctanh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)214 static void nnc_aten_arctanh(
215 int64_t bufs_num,
216 void** buf_data,
217 int64_t* buf_ranks,
218 int64_t* buf_dims,
219 int64_t* buf_strides,
220 int8_t* buf_dtypes,
221 int64_t args_num,
222 int64_t* extra_args) {
223 std::vector<at::Tensor> tensors = constructTensors(
224 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
225 at::Tensor& r = tensors[0];
226 const at::Tensor& self = tensors[1];
227 try {
228 at::arctanh_out(r, self);
229 } catch (...) {
230 }
231 }
nnc_aten_asin(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)232 static void nnc_aten_asin(
233 int64_t bufs_num,
234 void** buf_data,
235 int64_t* buf_ranks,
236 int64_t* buf_dims,
237 int64_t* buf_strides,
238 int8_t* buf_dtypes,
239 int64_t args_num,
240 int64_t* extra_args) {
241 std::vector<at::Tensor> tensors = constructTensors(
242 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
243 at::Tensor& r = tensors[0];
244 const at::Tensor& self = tensors[1];
245 try {
246 at::asin_out(r, self);
247 } catch (...) {
248 }
249 }
nnc_aten_arcsin(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)250 static void nnc_aten_arcsin(
251 int64_t bufs_num,
252 void** buf_data,
253 int64_t* buf_ranks,
254 int64_t* buf_dims,
255 int64_t* buf_strides,
256 int8_t* buf_dtypes,
257 int64_t args_num,
258 int64_t* extra_args) {
259 std::vector<at::Tensor> tensors = constructTensors(
260 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
261 at::Tensor& r = tensors[0];
262 const at::Tensor& self = tensors[1];
263 try {
264 at::arcsin_out(r, self);
265 } catch (...) {
266 }
267 }
nnc_aten_atan(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)268 static void nnc_aten_atan(
269 int64_t bufs_num,
270 void** buf_data,
271 int64_t* buf_ranks,
272 int64_t* buf_dims,
273 int64_t* buf_strides,
274 int8_t* buf_dtypes,
275 int64_t args_num,
276 int64_t* extra_args) {
277 std::vector<at::Tensor> tensors = constructTensors(
278 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
279 at::Tensor& r = tensors[0];
280 const at::Tensor& self = tensors[1];
281 try {
282 at::atan_out(r, self);
283 } catch (...) {
284 }
285 }
nnc_aten_arctan(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)286 static void nnc_aten_arctan(
287 int64_t bufs_num,
288 void** buf_data,
289 int64_t* buf_ranks,
290 int64_t* buf_dims,
291 int64_t* buf_strides,
292 int8_t* buf_dtypes,
293 int64_t args_num,
294 int64_t* extra_args) {
295 std::vector<at::Tensor> tensors = constructTensors(
296 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
297 at::Tensor& r = tensors[0];
298 const at::Tensor& self = tensors[1];
299 try {
300 at::arctan_out(r, self);
301 } catch (...) {
302 }
303 }
nnc_aten_bitwise_not(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)304 static void nnc_aten_bitwise_not(
305 int64_t bufs_num,
306 void** buf_data,
307 int64_t* buf_ranks,
308 int64_t* buf_dims,
309 int64_t* buf_strides,
310 int8_t* buf_dtypes,
311 int64_t args_num,
312 int64_t* extra_args) {
313 std::vector<at::Tensor> tensors = constructTensors(
314 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
315 at::Tensor& r = tensors[0];
316 const at::Tensor& self = tensors[1];
317 try {
318 at::bitwise_not_out(r, self);
319 } catch (...) {
320 }
321 }
nnc_aten_copysign(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)322 static void nnc_aten_copysign(
323 int64_t bufs_num,
324 void** buf_data,
325 int64_t* buf_ranks,
326 int64_t* buf_dims,
327 int64_t* buf_strides,
328 int8_t* buf_dtypes,
329 int64_t args_num,
330 int64_t* extra_args) {
331 std::vector<at::Tensor> tensors = constructTensors(
332 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
333 at::Tensor& r = tensors[0];
334 const at::Tensor& self = tensors[1];
335 const at::Tensor& other = tensors[2];
336 try {
337 at::copysign_out(r, self, other);
338 } catch (...) {
339 }
340 }
nnc_aten_logical_not(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)341 static void nnc_aten_logical_not(
342 int64_t bufs_num,
343 void** buf_data,
344 int64_t* buf_ranks,
345 int64_t* buf_dims,
346 int64_t* buf_strides,
347 int8_t* buf_dtypes,
348 int64_t args_num,
349 int64_t* extra_args) {
350 std::vector<at::Tensor> tensors = constructTensors(
351 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
352 at::Tensor& r = tensors[0];
353 const at::Tensor& self = tensors[1];
354 try {
355 at::logical_not_out(r, self);
356 } catch (...) {
357 }
358 }
nnc_aten_logical_xor(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)359 static void nnc_aten_logical_xor(
360 int64_t bufs_num,
361 void** buf_data,
362 int64_t* buf_ranks,
363 int64_t* buf_dims,
364 int64_t* buf_strides,
365 int8_t* buf_dtypes,
366 int64_t args_num,
367 int64_t* extra_args) {
368 std::vector<at::Tensor> tensors = constructTensors(
369 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
370 at::Tensor& r = tensors[0];
371 const at::Tensor& self = tensors[1];
372 const at::Tensor& other = tensors[2];
373 try {
374 at::logical_xor_out(r, self, other);
375 } catch (...) {
376 }
377 }
nnc_aten_logical_and(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)378 static void nnc_aten_logical_and(
379 int64_t bufs_num,
380 void** buf_data,
381 int64_t* buf_ranks,
382 int64_t* buf_dims,
383 int64_t* buf_strides,
384 int8_t* buf_dtypes,
385 int64_t args_num,
386 int64_t* extra_args) {
387 std::vector<at::Tensor> tensors = constructTensors(
388 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
389 at::Tensor& r = tensors[0];
390 const at::Tensor& self = tensors[1];
391 const at::Tensor& other = tensors[2];
392 try {
393 at::logical_and_out(r, self, other);
394 } catch (...) {
395 }
396 }
nnc_aten_logical_or(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)397 static void nnc_aten_logical_or(
398 int64_t bufs_num,
399 void** buf_data,
400 int64_t* buf_ranks,
401 int64_t* buf_dims,
402 int64_t* buf_strides,
403 int8_t* buf_dtypes,
404 int64_t args_num,
405 int64_t* extra_args) {
406 std::vector<at::Tensor> tensors = constructTensors(
407 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
408 at::Tensor& r = tensors[0];
409 const at::Tensor& self = tensors[1];
410 const at::Tensor& other = tensors[2];
411 try {
412 at::logical_or_out(r, self, other);
413 } catch (...) {
414 }
415 }
nnc_aten_bmm(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)416 static void nnc_aten_bmm(
417 int64_t bufs_num,
418 void** buf_data,
419 int64_t* buf_ranks,
420 int64_t* buf_dims,
421 int64_t* buf_strides,
422 int8_t* buf_dtypes,
423 int64_t args_num,
424 int64_t* extra_args) {
425 std::vector<at::Tensor> tensors = constructTensors(
426 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
427 at::Tensor& r = tensors[0];
428 const at::Tensor& self = tensors[1];
429 const at::Tensor& mat2 = tensors[2];
430 try {
431 at::bmm_out(r, self, mat2);
432 } catch (...) {
433 }
434 }
nnc_aten_ceil(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)435 static void nnc_aten_ceil(
436 int64_t bufs_num,
437 void** buf_data,
438 int64_t* buf_ranks,
439 int64_t* buf_dims,
440 int64_t* buf_strides,
441 int8_t* buf_dtypes,
442 int64_t args_num,
443 int64_t* extra_args) {
444 std::vector<at::Tensor> tensors = constructTensors(
445 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
446 at::Tensor& r = tensors[0];
447 const at::Tensor& self = tensors[1];
448 try {
449 at::ceil_out(r, self);
450 } catch (...) {
451 }
452 }
nnc_aten_clamp_max(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)453 static void nnc_aten_clamp_max(
454 int64_t bufs_num,
455 void** buf_data,
456 int64_t* buf_ranks,
457 int64_t* buf_dims,
458 int64_t* buf_strides,
459 int8_t* buf_dtypes,
460 int64_t args_num,
461 int64_t* extra_args) {
462 std::vector<at::Tensor> tensors = constructTensors(
463 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
464 at::Tensor& r = tensors[0];
465 const at::Tensor& self = tensors[1];
466 const at::Tensor& max = tensors[2];
467 try {
468 at::clamp_max_out(r, self, max);
469 } catch (...) {
470 }
471 }
nnc_aten_clamp_min(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)472 static void nnc_aten_clamp_min(
473 int64_t bufs_num,
474 void** buf_data,
475 int64_t* buf_ranks,
476 int64_t* buf_dims,
477 int64_t* buf_strides,
478 int8_t* buf_dtypes,
479 int64_t args_num,
480 int64_t* extra_args) {
481 std::vector<at::Tensor> tensors = constructTensors(
482 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
483 at::Tensor& r = tensors[0];
484 const at::Tensor& self = tensors[1];
485 const at::Tensor& min = tensors[2];
486 try {
487 at::clamp_min_out(r, self, min);
488 } catch (...) {
489 }
490 }
nnc_aten_complex(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)491 static void nnc_aten_complex(
492 int64_t bufs_num,
493 void** buf_data,
494 int64_t* buf_ranks,
495 int64_t* buf_dims,
496 int64_t* buf_strides,
497 int8_t* buf_dtypes,
498 int64_t args_num,
499 int64_t* extra_args) {
500 std::vector<at::Tensor> tensors = constructTensors(
501 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
502 at::Tensor& r = tensors[0];
503 const at::Tensor& real = tensors[1];
504 const at::Tensor& imag = tensors[2];
505 try {
506 at::complex_out(r, real, imag);
507 } catch (...) {
508 }
509 }
nnc_aten_polar(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)510 static void nnc_aten_polar(
511 int64_t bufs_num,
512 void** buf_data,
513 int64_t* buf_ranks,
514 int64_t* buf_dims,
515 int64_t* buf_strides,
516 int8_t* buf_dtypes,
517 int64_t args_num,
518 int64_t* extra_args) {
519 std::vector<at::Tensor> tensors = constructTensors(
520 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
521 at::Tensor& r = tensors[0];
522 const at::Tensor& abs = tensors[1];
523 const at::Tensor& angle = tensors[2];
524 try {
525 at::polar_out(r, abs, angle);
526 } catch (...) {
527 }
528 }
nnc_aten_cos(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)529 static void nnc_aten_cos(
530 int64_t bufs_num,
531 void** buf_data,
532 int64_t* buf_ranks,
533 int64_t* buf_dims,
534 int64_t* buf_strides,
535 int8_t* buf_dtypes,
536 int64_t args_num,
537 int64_t* extra_args) {
538 std::vector<at::Tensor> tensors = constructTensors(
539 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
540 at::Tensor& r = tensors[0];
541 const at::Tensor& self = tensors[1];
542 try {
543 at::cos_out(r, self);
544 } catch (...) {
545 }
546 }
nnc_aten_cosh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)547 static void nnc_aten_cosh(
548 int64_t bufs_num,
549 void** buf_data,
550 int64_t* buf_ranks,
551 int64_t* buf_dims,
552 int64_t* buf_strides,
553 int8_t* buf_dtypes,
554 int64_t args_num,
555 int64_t* extra_args) {
556 std::vector<at::Tensor> tensors = constructTensors(
557 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
558 at::Tensor& r = tensors[0];
559 const at::Tensor& self = tensors[1];
560 try {
561 at::cosh_out(r, self);
562 } catch (...) {
563 }
564 }
nnc_aten_div(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)565 static void nnc_aten_div(
566 int64_t bufs_num,
567 void** buf_data,
568 int64_t* buf_ranks,
569 int64_t* buf_dims,
570 int64_t* buf_strides,
571 int8_t* buf_dtypes,
572 int64_t args_num,
573 int64_t* extra_args) {
574 std::vector<at::Tensor> tensors = constructTensors(
575 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
576 at::Tensor& r = tensors[0];
577 const at::Tensor& self = tensors[1];
578 const at::Tensor& other = tensors[2];
579 try {
580 at::div_out(r, self, other);
581 } catch (...) {
582 }
583 }
nnc_aten_divide(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)584 static void nnc_aten_divide(
585 int64_t bufs_num,
586 void** buf_data,
587 int64_t* buf_ranks,
588 int64_t* buf_dims,
589 int64_t* buf_strides,
590 int8_t* buf_dtypes,
591 int64_t args_num,
592 int64_t* extra_args) {
593 std::vector<at::Tensor> tensors = constructTensors(
594 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
595 at::Tensor& r = tensors[0];
596 const at::Tensor& self = tensors[1];
597 const at::Tensor& other = tensors[2];
598 try {
599 at::divide_out(r, self, other);
600 } catch (...) {
601 }
602 }
nnc_aten_true_divide(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)603 static void nnc_aten_true_divide(
604 int64_t bufs_num,
605 void** buf_data,
606 int64_t* buf_ranks,
607 int64_t* buf_dims,
608 int64_t* buf_strides,
609 int8_t* buf_dtypes,
610 int64_t args_num,
611 int64_t* extra_args) {
612 std::vector<at::Tensor> tensors = constructTensors(
613 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
614 at::Tensor& r = tensors[0];
615 const at::Tensor& self = tensors[1];
616 const at::Tensor& other = tensors[2];
617 try {
618 at::true_divide_out(r, self, other);
619 } catch (...) {
620 }
621 }
nnc_aten_dot(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)622 static void nnc_aten_dot(
623 int64_t bufs_num,
624 void** buf_data,
625 int64_t* buf_ranks,
626 int64_t* buf_dims,
627 int64_t* buf_strides,
628 int8_t* buf_dtypes,
629 int64_t args_num,
630 int64_t* extra_args) {
631 std::vector<at::Tensor> tensors = constructTensors(
632 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
633 at::Tensor& r = tensors[0];
634 const at::Tensor& self = tensors[1];
635 const at::Tensor& tensor = tensors[2];
636 try {
637 at::dot_out(r, self, tensor);
638 } catch (...) {
639 }
640 }
nnc_aten_vdot(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)641 static void nnc_aten_vdot(
642 int64_t bufs_num,
643 void** buf_data,
644 int64_t* buf_ranks,
645 int64_t* buf_dims,
646 int64_t* buf_strides,
647 int8_t* buf_dtypes,
648 int64_t args_num,
649 int64_t* extra_args) {
650 std::vector<at::Tensor> tensors = constructTensors(
651 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
652 at::Tensor& r = tensors[0];
653 const at::Tensor& self = tensors[1];
654 const at::Tensor& other = tensors[2];
655 try {
656 at::vdot_out(r, self, other);
657 } catch (...) {
658 }
659 }
nnc_aten_erf(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)660 static void nnc_aten_erf(
661 int64_t bufs_num,
662 void** buf_data,
663 int64_t* buf_ranks,
664 int64_t* buf_dims,
665 int64_t* buf_strides,
666 int8_t* buf_dtypes,
667 int64_t args_num,
668 int64_t* extra_args) {
669 std::vector<at::Tensor> tensors = constructTensors(
670 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
671 at::Tensor& r = tensors[0];
672 const at::Tensor& self = tensors[1];
673 try {
674 at::erf_out(r, self);
675 } catch (...) {
676 }
677 }
nnc_aten_erfc(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)678 static void nnc_aten_erfc(
679 int64_t bufs_num,
680 void** buf_data,
681 int64_t* buf_ranks,
682 int64_t* buf_dims,
683 int64_t* buf_strides,
684 int8_t* buf_dtypes,
685 int64_t args_num,
686 int64_t* extra_args) {
687 std::vector<at::Tensor> tensors = constructTensors(
688 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
689 at::Tensor& r = tensors[0];
690 const at::Tensor& self = tensors[1];
691 try {
692 at::erfc_out(r, self);
693 } catch (...) {
694 }
695 }
nnc_aten_exp(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)696 static void nnc_aten_exp(
697 int64_t bufs_num,
698 void** buf_data,
699 int64_t* buf_ranks,
700 int64_t* buf_dims,
701 int64_t* buf_strides,
702 int8_t* buf_dtypes,
703 int64_t args_num,
704 int64_t* extra_args) {
705 std::vector<at::Tensor> tensors = constructTensors(
706 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
707 at::Tensor& r = tensors[0];
708 const at::Tensor& self = tensors[1];
709 try {
710 at::exp_out(r, self);
711 } catch (...) {
712 }
713 }
nnc_aten_exp2(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)714 static void nnc_aten_exp2(
715 int64_t bufs_num,
716 void** buf_data,
717 int64_t* buf_ranks,
718 int64_t* buf_dims,
719 int64_t* buf_strides,
720 int8_t* buf_dtypes,
721 int64_t args_num,
722 int64_t* extra_args) {
723 std::vector<at::Tensor> tensors = constructTensors(
724 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
725 at::Tensor& r = tensors[0];
726 const at::Tensor& self = tensors[1];
727 try {
728 at::exp2_out(r, self);
729 } catch (...) {
730 }
731 }
nnc_aten_expm1(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)732 static void nnc_aten_expm1(
733 int64_t bufs_num,
734 void** buf_data,
735 int64_t* buf_ranks,
736 int64_t* buf_dims,
737 int64_t* buf_strides,
738 int8_t* buf_dtypes,
739 int64_t args_num,
740 int64_t* extra_args) {
741 std::vector<at::Tensor> tensors = constructTensors(
742 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
743 at::Tensor& r = tensors[0];
744 const at::Tensor& self = tensors[1];
745 try {
746 at::expm1_out(r, self);
747 } catch (...) {
748 }
749 }
nnc_aten_floor(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)750 static void nnc_aten_floor(
751 int64_t bufs_num,
752 void** buf_data,
753 int64_t* buf_ranks,
754 int64_t* buf_dims,
755 int64_t* buf_strides,
756 int8_t* buf_dtypes,
757 int64_t args_num,
758 int64_t* extra_args) {
759 std::vector<at::Tensor> tensors = constructTensors(
760 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
761 at::Tensor& r = tensors[0];
762 const at::Tensor& self = tensors[1];
763 try {
764 at::floor_out(r, self);
765 } catch (...) {
766 }
767 }
nnc_aten_floor_divide(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)768 static void nnc_aten_floor_divide(
769 int64_t bufs_num,
770 void** buf_data,
771 int64_t* buf_ranks,
772 int64_t* buf_dims,
773 int64_t* buf_strides,
774 int8_t* buf_dtypes,
775 int64_t args_num,
776 int64_t* extra_args) {
777 std::vector<at::Tensor> tensors = constructTensors(
778 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
779 at::Tensor& r = tensors[0];
780 const at::Tensor& self = tensors[1];
781 const at::Tensor& other = tensors[2];
782 try {
783 at::floor_divide_out(r, self, other);
784 } catch (...) {
785 }
786 }
nnc_aten_frac(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)787 static void nnc_aten_frac(
788 int64_t bufs_num,
789 void** buf_data,
790 int64_t* buf_ranks,
791 int64_t* buf_dims,
792 int64_t* buf_strides,
793 int8_t* buf_dtypes,
794 int64_t args_num,
795 int64_t* extra_args) {
796 std::vector<at::Tensor> tensors = constructTensors(
797 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
798 at::Tensor& r = tensors[0];
799 const at::Tensor& self = tensors[1];
800 try {
801 at::frac_out(r, self);
802 } catch (...) {
803 }
804 }
nnc_aten_gcd(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)805 static void nnc_aten_gcd(
806 int64_t bufs_num,
807 void** buf_data,
808 int64_t* buf_ranks,
809 int64_t* buf_dims,
810 int64_t* buf_strides,
811 int8_t* buf_dtypes,
812 int64_t args_num,
813 int64_t* extra_args) {
814 std::vector<at::Tensor> tensors = constructTensors(
815 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
816 at::Tensor& r = tensors[0];
817 const at::Tensor& self = tensors[1];
818 const at::Tensor& other = tensors[2];
819 try {
820 at::gcd_out(r, self, other);
821 } catch (...) {
822 }
823 }
nnc_aten_lcm(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)824 static void nnc_aten_lcm(
825 int64_t bufs_num,
826 void** buf_data,
827 int64_t* buf_ranks,
828 int64_t* buf_dims,
829 int64_t* buf_strides,
830 int8_t* buf_dtypes,
831 int64_t args_num,
832 int64_t* extra_args) {
833 std::vector<at::Tensor> tensors = constructTensors(
834 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
835 at::Tensor& r = tensors[0];
836 const at::Tensor& self = tensors[1];
837 const at::Tensor& other = tensors[2];
838 try {
839 at::lcm_out(r, self, other);
840 } catch (...) {
841 }
842 }
nnc_aten_inverse(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)843 static void nnc_aten_inverse(
844 int64_t bufs_num,
845 void** buf_data,
846 int64_t* buf_ranks,
847 int64_t* buf_dims,
848 int64_t* buf_strides,
849 int8_t* buf_dtypes,
850 int64_t args_num,
851 int64_t* extra_args) {
852 std::vector<at::Tensor> tensors = constructTensors(
853 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
854 at::Tensor& r = tensors[0];
855 const at::Tensor& self = tensors[1];
856 try {
857 at::inverse_out(r, self);
858 } catch (...) {
859 }
860 }
nnc_aten_kron(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)861 static void nnc_aten_kron(
862 int64_t bufs_num,
863 void** buf_data,
864 int64_t* buf_ranks,
865 int64_t* buf_dims,
866 int64_t* buf_strides,
867 int8_t* buf_dtypes,
868 int64_t args_num,
869 int64_t* extra_args) {
870 std::vector<at::Tensor> tensors = constructTensors(
871 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
872 at::Tensor& r = tensors[0];
873 const at::Tensor& self = tensors[1];
874 const at::Tensor& other = tensors[2];
875 try {
876 at::kron_out(r, self, other);
877 } catch (...) {
878 }
879 }
nnc_aten_ldexp(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)880 static void nnc_aten_ldexp(
881 int64_t bufs_num,
882 void** buf_data,
883 int64_t* buf_ranks,
884 int64_t* buf_dims,
885 int64_t* buf_strides,
886 int8_t* buf_dtypes,
887 int64_t args_num,
888 int64_t* extra_args) {
889 std::vector<at::Tensor> tensors = constructTensors(
890 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
891 at::Tensor& r = tensors[0];
892 const at::Tensor& self = tensors[1];
893 const at::Tensor& other = tensors[2];
894 try {
895 at::ldexp_out(r, self, other);
896 } catch (...) {
897 }
898 }
nnc_aten_log(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)899 static void nnc_aten_log(
900 int64_t bufs_num,
901 void** buf_data,
902 int64_t* buf_ranks,
903 int64_t* buf_dims,
904 int64_t* buf_strides,
905 int8_t* buf_dtypes,
906 int64_t args_num,
907 int64_t* extra_args) {
908 std::vector<at::Tensor> tensors = constructTensors(
909 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
910 at::Tensor& r = tensors[0];
911 const at::Tensor& self = tensors[1];
912 try {
913 at::log_out(r, self);
914 } catch (...) {
915 }
916 }
nnc_aten_log10(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)917 static void nnc_aten_log10(
918 int64_t bufs_num,
919 void** buf_data,
920 int64_t* buf_ranks,
921 int64_t* buf_dims,
922 int64_t* buf_strides,
923 int8_t* buf_dtypes,
924 int64_t args_num,
925 int64_t* extra_args) {
926 std::vector<at::Tensor> tensors = constructTensors(
927 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
928 at::Tensor& r = tensors[0];
929 const at::Tensor& self = tensors[1];
930 try {
931 at::log10_out(r, self);
932 } catch (...) {
933 }
934 }
nnc_aten_log1p(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)935 static void nnc_aten_log1p(
936 int64_t bufs_num,
937 void** buf_data,
938 int64_t* buf_ranks,
939 int64_t* buf_dims,
940 int64_t* buf_strides,
941 int8_t* buf_dtypes,
942 int64_t args_num,
943 int64_t* extra_args) {
944 std::vector<at::Tensor> tensors = constructTensors(
945 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
946 at::Tensor& r = tensors[0];
947 const at::Tensor& self = tensors[1];
948 try {
949 at::log1p_out(r, self);
950 } catch (...) {
951 }
952 }
nnc_aten_log2(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)953 static void nnc_aten_log2(
954 int64_t bufs_num,
955 void** buf_data,
956 int64_t* buf_ranks,
957 int64_t* buf_dims,
958 int64_t* buf_strides,
959 int8_t* buf_dtypes,
960 int64_t args_num,
961 int64_t* extra_args) {
962 std::vector<at::Tensor> tensors = constructTensors(
963 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
964 at::Tensor& r = tensors[0];
965 const at::Tensor& self = tensors[1];
966 try {
967 at::log2_out(r, self);
968 } catch (...) {
969 }
970 }
nnc_aten_logaddexp(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)971 static void nnc_aten_logaddexp(
972 int64_t bufs_num,
973 void** buf_data,
974 int64_t* buf_ranks,
975 int64_t* buf_dims,
976 int64_t* buf_strides,
977 int8_t* buf_dtypes,
978 int64_t args_num,
979 int64_t* extra_args) {
980 std::vector<at::Tensor> tensors = constructTensors(
981 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
982 at::Tensor& r = tensors[0];
983 const at::Tensor& self = tensors[1];
984 const at::Tensor& other = tensors[2];
985 try {
986 at::logaddexp_out(r, self, other);
987 } catch (...) {
988 }
989 }
nnc_aten_logaddexp2(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)990 static void nnc_aten_logaddexp2(
991 int64_t bufs_num,
992 void** buf_data,
993 int64_t* buf_ranks,
994 int64_t* buf_dims,
995 int64_t* buf_strides,
996 int8_t* buf_dtypes,
997 int64_t args_num,
998 int64_t* extra_args) {
999 std::vector<at::Tensor> tensors = constructTensors(
1000 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1001 at::Tensor& r = tensors[0];
1002 const at::Tensor& self = tensors[1];
1003 const at::Tensor& other = tensors[2];
1004 try {
1005 at::logaddexp2_out(r, self, other);
1006 } catch (...) {
1007 }
1008 }
nnc_aten_xlogy(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1009 static void nnc_aten_xlogy(
1010 int64_t bufs_num,
1011 void** buf_data,
1012 int64_t* buf_ranks,
1013 int64_t* buf_dims,
1014 int64_t* buf_strides,
1015 int8_t* buf_dtypes,
1016 int64_t args_num,
1017 int64_t* extra_args) {
1018 std::vector<at::Tensor> tensors = constructTensors(
1019 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1020 at::Tensor& r = tensors[0];
1021 const at::Tensor& self = tensors[1];
1022 const at::Tensor& other = tensors[2];
1023 try {
1024 at::xlogy_out(r, self, other);
1025 } catch (...) {
1026 }
1027 }
nnc_aten_matmul(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1028 void nnc_aten_matmul(
1029 int64_t bufs_num,
1030 void** buf_data,
1031 int64_t* buf_ranks,
1032 int64_t* buf_dims,
1033 int64_t* buf_strides,
1034 int8_t* buf_dtypes,
1035 int64_t args_num,
1036 int64_t* extra_args) {
1037 std::vector<at::Tensor> tensors = constructTensors(
1038 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1039 at::Tensor& r = tensors[0];
1040 const at::Tensor& self = tensors[1];
1041 const at::Tensor& other = tensors[2];
1042 try {
1043 at::matmul_out(r, self, other);
1044 } catch (...) {
1045 }
1046 }
nnc_aten__compute_linear_combination(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1047 static void nnc_aten__compute_linear_combination(
1048 int64_t bufs_num,
1049 void** buf_data,
1050 int64_t* buf_ranks,
1051 int64_t* buf_dims,
1052 int64_t* buf_strides,
1053 int8_t* buf_dtypes,
1054 int64_t args_num,
1055 int64_t* extra_args) {
1056 std::vector<at::Tensor> tensors = constructTensors(
1057 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1058 at::Tensor& r = tensors[0];
1059 const at::Tensor& input = tensors[1];
1060 const at::Tensor& coefficients = tensors[2];
1061 try {
1062 at::_compute_linear_combination_out(r, input, coefficients);
1063 } catch (...) {
1064 }
1065 }
nnc_aten_mm(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1066 void nnc_aten_mm(
1067 int64_t bufs_num,
1068 void** buf_data,
1069 int64_t* buf_ranks,
1070 int64_t* buf_dims,
1071 int64_t* buf_strides,
1072 int8_t* buf_dtypes,
1073 int64_t args_num,
1074 int64_t* extra_args) {
1075 std::vector<at::Tensor> tensors = constructTensors(
1076 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1077 at::Tensor& r = tensors[0];
1078 const at::Tensor& self = tensors[1];
1079 const at::Tensor& mat2 = tensors[2];
1080 try {
1081 at::mm_out(r, self, mat2);
1082 } catch (...) {
1083 }
1084 }
nnc_aten_mul(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1085 static void nnc_aten_mul(
1086 int64_t bufs_num,
1087 void** buf_data,
1088 int64_t* buf_ranks,
1089 int64_t* buf_dims,
1090 int64_t* buf_strides,
1091 int8_t* buf_dtypes,
1092 int64_t args_num,
1093 int64_t* extra_args) {
1094 std::vector<at::Tensor> tensors = constructTensors(
1095 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1096 at::Tensor& r = tensors[0];
1097 const at::Tensor& self = tensors[1];
1098 const at::Tensor& other = tensors[2];
1099 try {
1100 at::mul_out(r, self, other);
1101 } catch (...) {
1102 }
1103 }
nnc_aten_multiply(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1104 static void nnc_aten_multiply(
1105 int64_t bufs_num,
1106 void** buf_data,
1107 int64_t* buf_ranks,
1108 int64_t* buf_dims,
1109 int64_t* buf_strides,
1110 int8_t* buf_dtypes,
1111 int64_t args_num,
1112 int64_t* extra_args) {
1113 std::vector<at::Tensor> tensors = constructTensors(
1114 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1115 at::Tensor& r = tensors[0];
1116 const at::Tensor& self = tensors[1];
1117 const at::Tensor& other = tensors[2];
1118 try {
1119 at::multiply_out(r, self, other);
1120 } catch (...) {
1121 }
1122 }
nnc_aten_mv(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1123 void nnc_aten_mv(
1124 int64_t bufs_num,
1125 void** buf_data,
1126 int64_t* buf_ranks,
1127 int64_t* buf_dims,
1128 int64_t* buf_strides,
1129 int8_t* buf_dtypes,
1130 int64_t args_num,
1131 int64_t* extra_args) {
1132 std::vector<at::Tensor> tensors = constructTensors(
1133 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1134 at::Tensor& r = tensors[0];
1135 const at::Tensor& self = tensors[1];
1136 const at::Tensor& vec = tensors[2];
1137 try {
1138 at::mv_out(r, self, vec);
1139 } catch (...) {
1140 }
1141 }
nnc_aten_rad2deg(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1142 static void nnc_aten_rad2deg(
1143 int64_t bufs_num,
1144 void** buf_data,
1145 int64_t* buf_ranks,
1146 int64_t* buf_dims,
1147 int64_t* buf_strides,
1148 int8_t* buf_dtypes,
1149 int64_t args_num,
1150 int64_t* extra_args) {
1151 std::vector<at::Tensor> tensors = constructTensors(
1152 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1153 at::Tensor& r = tensors[0];
1154 const at::Tensor& self = tensors[1];
1155 try {
1156 at::rad2deg_out(r, self);
1157 } catch (...) {
1158 }
1159 }
nnc_aten_deg2rad(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1160 static void nnc_aten_deg2rad(
1161 int64_t bufs_num,
1162 void** buf_data,
1163 int64_t* buf_ranks,
1164 int64_t* buf_dims,
1165 int64_t* buf_strides,
1166 int8_t* buf_dtypes,
1167 int64_t args_num,
1168 int64_t* extra_args) {
1169 std::vector<at::Tensor> tensors = constructTensors(
1170 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1171 at::Tensor& r = tensors[0];
1172 const at::Tensor& self = tensors[1];
1173 try {
1174 at::deg2rad_out(r, self);
1175 } catch (...) {
1176 }
1177 }
nnc_aten_reciprocal(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1178 static void nnc_aten_reciprocal(
1179 int64_t bufs_num,
1180 void** buf_data,
1181 int64_t* buf_ranks,
1182 int64_t* buf_dims,
1183 int64_t* buf_strides,
1184 int8_t* buf_dtypes,
1185 int64_t args_num,
1186 int64_t* extra_args) {
1187 std::vector<at::Tensor> tensors = constructTensors(
1188 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1189 at::Tensor& r = tensors[0];
1190 const at::Tensor& self = tensors[1];
1191 try {
1192 at::reciprocal_out(r, self);
1193 } catch (...) {
1194 }
1195 }
nnc_aten_neg(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1196 static void nnc_aten_neg(
1197 int64_t bufs_num,
1198 void** buf_data,
1199 int64_t* buf_ranks,
1200 int64_t* buf_dims,
1201 int64_t* buf_strides,
1202 int8_t* buf_dtypes,
1203 int64_t args_num,
1204 int64_t* extra_args) {
1205 std::vector<at::Tensor> tensors = constructTensors(
1206 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1207 at::Tensor& r = tensors[0];
1208 const at::Tensor& self = tensors[1];
1209 try {
1210 at::neg_out(r, self);
1211 } catch (...) {
1212 }
1213 }
nnc_aten_negative(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1214 static void nnc_aten_negative(
1215 int64_t bufs_num,
1216 void** buf_data,
1217 int64_t* buf_ranks,
1218 int64_t* buf_dims,
1219 int64_t* buf_strides,
1220 int8_t* buf_dtypes,
1221 int64_t args_num,
1222 int64_t* extra_args) {
1223 std::vector<at::Tensor> tensors = constructTensors(
1224 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1225 at::Tensor& r = tensors[0];
1226 const at::Tensor& self = tensors[1];
1227 try {
1228 at::negative_out(r, self);
1229 } catch (...) {
1230 }
1231 }
nnc_aten_round(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1232 static void nnc_aten_round(
1233 int64_t bufs_num,
1234 void** buf_data,
1235 int64_t* buf_ranks,
1236 int64_t* buf_dims,
1237 int64_t* buf_strides,
1238 int8_t* buf_dtypes,
1239 int64_t args_num,
1240 int64_t* extra_args) {
1241 std::vector<at::Tensor> tensors = constructTensors(
1242 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1243 at::Tensor& r = tensors[0];
1244 const at::Tensor& self = tensors[1];
1245 try {
1246 at::round_out(r, self);
1247 } catch (...) {
1248 }
1249 }
nnc_aten_rsqrt(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1250 static void nnc_aten_rsqrt(
1251 int64_t bufs_num,
1252 void** buf_data,
1253 int64_t* buf_ranks,
1254 int64_t* buf_dims,
1255 int64_t* buf_strides,
1256 int8_t* buf_dtypes,
1257 int64_t args_num,
1258 int64_t* extra_args) {
1259 std::vector<at::Tensor> tensors = constructTensors(
1260 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1261 at::Tensor& r = tensors[0];
1262 const at::Tensor& self = tensors[1];
1263 try {
1264 at::rsqrt_out(r, self);
1265 } catch (...) {
1266 }
1267 }
nnc_aten_silu(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1268 static void nnc_aten_silu(
1269 int64_t bufs_num,
1270 void** buf_data,
1271 int64_t* buf_ranks,
1272 int64_t* buf_dims,
1273 int64_t* buf_strides,
1274 int8_t* buf_dtypes,
1275 int64_t args_num,
1276 int64_t* extra_args) {
1277 std::vector<at::Tensor> tensors = constructTensors(
1278 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1279 at::Tensor& r = tensors[0];
1280 const at::Tensor& self = tensors[1];
1281 try {
1282 at::silu_out(r, self);
1283 } catch (...) {
1284 }
1285 }
nnc_aten_mish(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1286 static void nnc_aten_mish(
1287 int64_t bufs_num,
1288 void** buf_data,
1289 int64_t* buf_ranks,
1290 int64_t* buf_dims,
1291 int64_t* buf_strides,
1292 int8_t* buf_dtypes,
1293 int64_t args_num,
1294 int64_t* extra_args) {
1295 std::vector<at::Tensor> tensors = constructTensors(
1296 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1297 at::Tensor& r = tensors[0];
1298 const at::Tensor& self = tensors[1];
1299 try {
1300 at::mish_out(r, self);
1301 } catch (...) {
1302 }
1303 }
nnc_aten_sigmoid(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1304 static void nnc_aten_sigmoid(
1305 int64_t bufs_num,
1306 void** buf_data,
1307 int64_t* buf_ranks,
1308 int64_t* buf_dims,
1309 int64_t* buf_strides,
1310 int8_t* buf_dtypes,
1311 int64_t args_num,
1312 int64_t* extra_args) {
1313 std::vector<at::Tensor> tensors = constructTensors(
1314 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1315 at::Tensor& r = tensors[0];
1316 const at::Tensor& self = tensors[1];
1317 try {
1318 at::sigmoid_out(r, self);
1319 } catch (...) {
1320 }
1321 }
nnc_aten_sin(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1322 static void nnc_aten_sin(
1323 int64_t bufs_num,
1324 void** buf_data,
1325 int64_t* buf_ranks,
1326 int64_t* buf_dims,
1327 int64_t* buf_strides,
1328 int8_t* buf_dtypes,
1329 int64_t args_num,
1330 int64_t* extra_args) {
1331 std::vector<at::Tensor> tensors = constructTensors(
1332 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1333 at::Tensor& r = tensors[0];
1334 const at::Tensor& self = tensors[1];
1335 try {
1336 at::sin_out(r, self);
1337 } catch (...) {
1338 }
1339 }
nnc_aten_sinc(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1340 static void nnc_aten_sinc(
1341 int64_t bufs_num,
1342 void** buf_data,
1343 int64_t* buf_ranks,
1344 int64_t* buf_dims,
1345 int64_t* buf_strides,
1346 int8_t* buf_dtypes,
1347 int64_t args_num,
1348 int64_t* extra_args) {
1349 std::vector<at::Tensor> tensors = constructTensors(
1350 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1351 at::Tensor& r = tensors[0];
1352 const at::Tensor& self = tensors[1];
1353 try {
1354 at::sinc_out(r, self);
1355 } catch (...) {
1356 }
1357 }
nnc_aten_sinh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1358 static void nnc_aten_sinh(
1359 int64_t bufs_num,
1360 void** buf_data,
1361 int64_t* buf_ranks,
1362 int64_t* buf_dims,
1363 int64_t* buf_strides,
1364 int8_t* buf_dtypes,
1365 int64_t args_num,
1366 int64_t* extra_args) {
1367 std::vector<at::Tensor> tensors = constructTensors(
1368 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1369 at::Tensor& r = tensors[0];
1370 const at::Tensor& self = tensors[1];
1371 try {
1372 at::sinh_out(r, self);
1373 } catch (...) {
1374 }
1375 }
nnc_aten_sqrt(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1376 static void nnc_aten_sqrt(
1377 int64_t bufs_num,
1378 void** buf_data,
1379 int64_t* buf_ranks,
1380 int64_t* buf_dims,
1381 int64_t* buf_strides,
1382 int8_t* buf_dtypes,
1383 int64_t args_num,
1384 int64_t* extra_args) {
1385 std::vector<at::Tensor> tensors = constructTensors(
1386 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1387 at::Tensor& r = tensors[0];
1388 const at::Tensor& self = tensors[1];
1389 try {
1390 at::sqrt_out(r, self);
1391 } catch (...) {
1392 }
1393 }
nnc_aten_square(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1394 static void nnc_aten_square(
1395 int64_t bufs_num,
1396 void** buf_data,
1397 int64_t* buf_ranks,
1398 int64_t* buf_dims,
1399 int64_t* buf_strides,
1400 int8_t* buf_dtypes,
1401 int64_t args_num,
1402 int64_t* extra_args) {
1403 std::vector<at::Tensor> tensors = constructTensors(
1404 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1405 at::Tensor& r = tensors[0];
1406 const at::Tensor& self = tensors[1];
1407 try {
1408 at::square_out(r, self);
1409 } catch (...) {
1410 }
1411 }
nnc_aten_tan(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1412 static void nnc_aten_tan(
1413 int64_t bufs_num,
1414 void** buf_data,
1415 int64_t* buf_ranks,
1416 int64_t* buf_dims,
1417 int64_t* buf_strides,
1418 int8_t* buf_dtypes,
1419 int64_t args_num,
1420 int64_t* extra_args) {
1421 std::vector<at::Tensor> tensors = constructTensors(
1422 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1423 at::Tensor& r = tensors[0];
1424 const at::Tensor& self = tensors[1];
1425 try {
1426 at::tan_out(r, self);
1427 } catch (...) {
1428 }
1429 }
nnc_aten_tanh(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1430 static void nnc_aten_tanh(
1431 int64_t bufs_num,
1432 void** buf_data,
1433 int64_t* buf_ranks,
1434 int64_t* buf_dims,
1435 int64_t* buf_strides,
1436 int8_t* buf_dtypes,
1437 int64_t args_num,
1438 int64_t* extra_args) {
1439 std::vector<at::Tensor> tensors = constructTensors(
1440 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1441 at::Tensor& r = tensors[0];
1442 const at::Tensor& self = tensors[1];
1443 try {
1444 at::tanh_out(r, self);
1445 } catch (...) {
1446 }
1447 }
nnc_aten_trunc(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1448 static void nnc_aten_trunc(
1449 int64_t bufs_num,
1450 void** buf_data,
1451 int64_t* buf_ranks,
1452 int64_t* buf_dims,
1453 int64_t* buf_strides,
1454 int8_t* buf_dtypes,
1455 int64_t args_num,
1456 int64_t* extra_args) {
1457 std::vector<at::Tensor> tensors = constructTensors(
1458 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1459 at::Tensor& r = tensors[0];
1460 const at::Tensor& self = tensors[1];
1461 try {
1462 at::trunc_out(r, self);
1463 } catch (...) {
1464 }
1465 }
nnc_aten_fix(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1466 static void nnc_aten_fix(
1467 int64_t bufs_num,
1468 void** buf_data,
1469 int64_t* buf_ranks,
1470 int64_t* buf_dims,
1471 int64_t* buf_strides,
1472 int8_t* buf_dtypes,
1473 int64_t args_num,
1474 int64_t* extra_args) {
1475 std::vector<at::Tensor> tensors = constructTensors(
1476 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1477 at::Tensor& r = tensors[0];
1478 const at::Tensor& self = tensors[1];
1479 try {
1480 at::fix_out(r, self);
1481 } catch (...) {
1482 }
1483 }
nnc_aten_heaviside(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1484 static void nnc_aten_heaviside(
1485 int64_t bufs_num,
1486 void** buf_data,
1487 int64_t* buf_ranks,
1488 int64_t* buf_dims,
1489 int64_t* buf_strides,
1490 int8_t* buf_dtypes,
1491 int64_t args_num,
1492 int64_t* extra_args) {
1493 std::vector<at::Tensor> tensors = constructTensors(
1494 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1495 at::Tensor& r = tensors[0];
1496 const at::Tensor& self = tensors[1];
1497 const at::Tensor& values = tensors[2];
1498 try {
1499 at::heaviside_out(r, self, values);
1500 } catch (...) {
1501 }
1502 }
nnc_aten_hspmm(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1503 static void nnc_aten_hspmm(
1504 int64_t bufs_num,
1505 void** buf_data,
1506 int64_t* buf_ranks,
1507 int64_t* buf_dims,
1508 int64_t* buf_strides,
1509 int8_t* buf_dtypes,
1510 int64_t args_num,
1511 int64_t* extra_args) {
1512 std::vector<at::Tensor> tensors = constructTensors(
1513 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1514 at::Tensor& r = tensors[0];
1515 const at::Tensor& mat1 = tensors[1];
1516 const at::Tensor& mat2 = tensors[2];
1517 try {
1518 at::hspmm_out(r, mat1, mat2);
1519 } catch (...) {
1520 }
1521 }
nnc_aten_bitwise_and(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1522 static void nnc_aten_bitwise_and(
1523 int64_t bufs_num,
1524 void** buf_data,
1525 int64_t* buf_ranks,
1526 int64_t* buf_dims,
1527 int64_t* buf_strides,
1528 int8_t* buf_dtypes,
1529 int64_t args_num,
1530 int64_t* extra_args) {
1531 std::vector<at::Tensor> tensors = constructTensors(
1532 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1533 at::Tensor& r = tensors[0];
1534 const at::Tensor& self = tensors[1];
1535 const at::Tensor& other = tensors[2];
1536 try {
1537 at::bitwise_and_out(r, self, other);
1538 } catch (...) {
1539 }
1540 }
nnc_aten_bitwise_or(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1541 static void nnc_aten_bitwise_or(
1542 int64_t bufs_num,
1543 void** buf_data,
1544 int64_t* buf_ranks,
1545 int64_t* buf_dims,
1546 int64_t* buf_strides,
1547 int8_t* buf_dtypes,
1548 int64_t args_num,
1549 int64_t* extra_args) {
1550 std::vector<at::Tensor> tensors = constructTensors(
1551 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1552 at::Tensor& r = tensors[0];
1553 const at::Tensor& self = tensors[1];
1554 const at::Tensor& other = tensors[2];
1555 try {
1556 at::bitwise_or_out(r, self, other);
1557 } catch (...) {
1558 }
1559 }
nnc_aten_bitwise_xor(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1560 static void nnc_aten_bitwise_xor(
1561 int64_t bufs_num,
1562 void** buf_data,
1563 int64_t* buf_ranks,
1564 int64_t* buf_dims,
1565 int64_t* buf_strides,
1566 int8_t* buf_dtypes,
1567 int64_t args_num,
1568 int64_t* extra_args) {
1569 std::vector<at::Tensor> tensors = constructTensors(
1570 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1571 at::Tensor& r = tensors[0];
1572 const at::Tensor& self = tensors[1];
1573 const at::Tensor& other = tensors[2];
1574 try {
1575 at::bitwise_xor_out(r, self, other);
1576 } catch (...) {
1577 }
1578 }
nnc_aten_ne(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1579 static void nnc_aten_ne(
1580 int64_t bufs_num,
1581 void** buf_data,
1582 int64_t* buf_ranks,
1583 int64_t* buf_dims,
1584 int64_t* buf_strides,
1585 int8_t* buf_dtypes,
1586 int64_t args_num,
1587 int64_t* extra_args) {
1588 std::vector<at::Tensor> tensors = constructTensors(
1589 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1590 at::Tensor& r = tensors[0];
1591 const at::Tensor& self = tensors[1];
1592 const at::Tensor& other = tensors[2];
1593 try {
1594 at::ne_out(r, self, other);
1595 } catch (...) {
1596 }
1597 }
nnc_aten_not_equal(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1598 static void nnc_aten_not_equal(
1599 int64_t bufs_num,
1600 void** buf_data,
1601 int64_t* buf_ranks,
1602 int64_t* buf_dims,
1603 int64_t* buf_strides,
1604 int8_t* buf_dtypes,
1605 int64_t args_num,
1606 int64_t* extra_args) {
1607 std::vector<at::Tensor> tensors = constructTensors(
1608 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1609 at::Tensor& r = tensors[0];
1610 const at::Tensor& self = tensors[1];
1611 const at::Tensor& other = tensors[2];
1612 try {
1613 at::not_equal_out(r, self, other);
1614 } catch (...) {
1615 }
1616 }
nnc_aten_eq(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1617 static void nnc_aten_eq(
1618 int64_t bufs_num,
1619 void** buf_data,
1620 int64_t* buf_ranks,
1621 int64_t* buf_dims,
1622 int64_t* buf_strides,
1623 int8_t* buf_dtypes,
1624 int64_t args_num,
1625 int64_t* extra_args) {
1626 std::vector<at::Tensor> tensors = constructTensors(
1627 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1628 at::Tensor& r = tensors[0];
1629 const at::Tensor& self = tensors[1];
1630 const at::Tensor& other = tensors[2];
1631 try {
1632 at::eq_out(r, self, other);
1633 } catch (...) {
1634 }
1635 }
nnc_aten_ge(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1636 static void nnc_aten_ge(
1637 int64_t bufs_num,
1638 void** buf_data,
1639 int64_t* buf_ranks,
1640 int64_t* buf_dims,
1641 int64_t* buf_strides,
1642 int8_t* buf_dtypes,
1643 int64_t args_num,
1644 int64_t* extra_args) {
1645 std::vector<at::Tensor> tensors = constructTensors(
1646 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1647 at::Tensor& r = tensors[0];
1648 const at::Tensor& self = tensors[1];
1649 const at::Tensor& other = tensors[2];
1650 try {
1651 at::ge_out(r, self, other);
1652 } catch (...) {
1653 }
1654 }
nnc_aten_greater_equal(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1655 static void nnc_aten_greater_equal(
1656 int64_t bufs_num,
1657 void** buf_data,
1658 int64_t* buf_ranks,
1659 int64_t* buf_dims,
1660 int64_t* buf_strides,
1661 int8_t* buf_dtypes,
1662 int64_t args_num,
1663 int64_t* extra_args) {
1664 std::vector<at::Tensor> tensors = constructTensors(
1665 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1666 at::Tensor& r = tensors[0];
1667 const at::Tensor& self = tensors[1];
1668 const at::Tensor& other = tensors[2];
1669 try {
1670 at::greater_equal_out(r, self, other);
1671 } catch (...) {
1672 }
1673 }
nnc_aten_le(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1674 static void nnc_aten_le(
1675 int64_t bufs_num,
1676 void** buf_data,
1677 int64_t* buf_ranks,
1678 int64_t* buf_dims,
1679 int64_t* buf_strides,
1680 int8_t* buf_dtypes,
1681 int64_t args_num,
1682 int64_t* extra_args) {
1683 std::vector<at::Tensor> tensors = constructTensors(
1684 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1685 at::Tensor& r = tensors[0];
1686 const at::Tensor& self = tensors[1];
1687 const at::Tensor& other = tensors[2];
1688 try {
1689 at::le_out(r, self, other);
1690 } catch (...) {
1691 }
1692 }
nnc_aten_less_equal(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1693 static void nnc_aten_less_equal(
1694 int64_t bufs_num,
1695 void** buf_data,
1696 int64_t* buf_ranks,
1697 int64_t* buf_dims,
1698 int64_t* buf_strides,
1699 int8_t* buf_dtypes,
1700 int64_t args_num,
1701 int64_t* extra_args) {
1702 std::vector<at::Tensor> tensors = constructTensors(
1703 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1704 at::Tensor& r = tensors[0];
1705 const at::Tensor& self = tensors[1];
1706 const at::Tensor& other = tensors[2];
1707 try {
1708 at::less_equal_out(r, self, other);
1709 } catch (...) {
1710 }
1711 }
nnc_aten_gt(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1712 static void nnc_aten_gt(
1713 int64_t bufs_num,
1714 void** buf_data,
1715 int64_t* buf_ranks,
1716 int64_t* buf_dims,
1717 int64_t* buf_strides,
1718 int8_t* buf_dtypes,
1719 int64_t args_num,
1720 int64_t* extra_args) {
1721 std::vector<at::Tensor> tensors = constructTensors(
1722 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1723 at::Tensor& r = tensors[0];
1724 const at::Tensor& self = tensors[1];
1725 const at::Tensor& other = tensors[2];
1726 try {
1727 at::gt_out(r, self, other);
1728 } catch (...) {
1729 }
1730 }
nnc_aten_greater(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1731 static void nnc_aten_greater(
1732 int64_t bufs_num,
1733 void** buf_data,
1734 int64_t* buf_ranks,
1735 int64_t* buf_dims,
1736 int64_t* buf_strides,
1737 int8_t* buf_dtypes,
1738 int64_t args_num,
1739 int64_t* extra_args) {
1740 std::vector<at::Tensor> tensors = constructTensors(
1741 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1742 at::Tensor& r = tensors[0];
1743 const at::Tensor& self = tensors[1];
1744 const at::Tensor& other = tensors[2];
1745 try {
1746 at::greater_out(r, self, other);
1747 } catch (...) {
1748 }
1749 }
nnc_aten_lt(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1750 static void nnc_aten_lt(
1751 int64_t bufs_num,
1752 void** buf_data,
1753 int64_t* buf_ranks,
1754 int64_t* buf_dims,
1755 int64_t* buf_strides,
1756 int8_t* buf_dtypes,
1757 int64_t args_num,
1758 int64_t* extra_args) {
1759 std::vector<at::Tensor> tensors = constructTensors(
1760 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1761 at::Tensor& r = tensors[0];
1762 const at::Tensor& self = tensors[1];
1763 const at::Tensor& other = tensors[2];
1764 try {
1765 at::lt_out(r, self, other);
1766 } catch (...) {
1767 }
1768 }
nnc_aten_less(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1769 static void nnc_aten_less(
1770 int64_t bufs_num,
1771 void** buf_data,
1772 int64_t* buf_ranks,
1773 int64_t* buf_dims,
1774 int64_t* buf_strides,
1775 int8_t* buf_dtypes,
1776 int64_t args_num,
1777 int64_t* extra_args) {
1778 std::vector<at::Tensor> tensors = constructTensors(
1779 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1780 at::Tensor& r = tensors[0];
1781 const at::Tensor& self = tensors[1];
1782 const at::Tensor& other = tensors[2];
1783 try {
1784 at::less_out(r, self, other);
1785 } catch (...) {
1786 }
1787 }
nnc_aten_take(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1788 static void nnc_aten_take(
1789 int64_t bufs_num,
1790 void** buf_data,
1791 int64_t* buf_ranks,
1792 int64_t* buf_dims,
1793 int64_t* buf_strides,
1794 int8_t* buf_dtypes,
1795 int64_t args_num,
1796 int64_t* extra_args) {
1797 std::vector<at::Tensor> tensors = constructTensors(
1798 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1799 at::Tensor& r = tensors[0];
1800 const at::Tensor& self = tensors[1];
1801 const at::Tensor& index = tensors[2];
1802 try {
1803 at::take_out(r, self, index);
1804 } catch (...) {
1805 }
1806 }
nnc_aten_masked_select(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1807 static void nnc_aten_masked_select(
1808 int64_t bufs_num,
1809 void** buf_data,
1810 int64_t* buf_ranks,
1811 int64_t* buf_dims,
1812 int64_t* buf_strides,
1813 int8_t* buf_dtypes,
1814 int64_t args_num,
1815 int64_t* extra_args) {
1816 std::vector<at::Tensor> tensors = constructTensors(
1817 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1818 at::Tensor& r = tensors[0];
1819 const at::Tensor& self = tensors[1];
1820 const at::Tensor& mask = tensors[2];
1821 try {
1822 at::masked_select_out(r, self, mask);
1823 } catch (...) {
1824 }
1825 }
nnc_aten_nonzero(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1826 static void nnc_aten_nonzero(
1827 int64_t bufs_num,
1828 void** buf_data,
1829 int64_t* buf_ranks,
1830 int64_t* buf_dims,
1831 int64_t* buf_strides,
1832 int8_t* buf_dtypes,
1833 int64_t args_num,
1834 int64_t* extra_args) {
1835 std::vector<at::Tensor> tensors = constructTensors(
1836 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1837 at::Tensor& r = tensors[0];
1838 const at::Tensor& self = tensors[1];
1839 try {
1840 at::nonzero_out(r, self);
1841 } catch (...) {
1842 }
1843 }
nnc_aten_orgqr(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1844 static void nnc_aten_orgqr(
1845 int64_t bufs_num,
1846 void** buf_data,
1847 int64_t* buf_ranks,
1848 int64_t* buf_dims,
1849 int64_t* buf_strides,
1850 int8_t* buf_dtypes,
1851 int64_t args_num,
1852 int64_t* extra_args) {
1853 std::vector<at::Tensor> tensors = constructTensors(
1854 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1855 at::Tensor& r = tensors[0];
1856 const at::Tensor& self = tensors[1];
1857 const at::Tensor& input2 = tensors[2];
1858 try {
1859 at::orgqr_out(r, self, input2);
1860 } catch (...) {
1861 }
1862 }
nnc_aten_lu_solve(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1863 static void nnc_aten_lu_solve(
1864 int64_t bufs_num,
1865 void** buf_data,
1866 int64_t* buf_ranks,
1867 int64_t* buf_dims,
1868 int64_t* buf_strides,
1869 int8_t* buf_dtypes,
1870 int64_t args_num,
1871 int64_t* extra_args) {
1872 std::vector<at::Tensor> tensors = constructTensors(
1873 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1874 at::Tensor& r = tensors[0];
1875 const at::Tensor& self = tensors[1];
1876 const at::Tensor& LU_data = tensors[2];
1877 const at::Tensor& LU_pivots = tensors[3];
1878 try {
1879 at::lu_solve_out(r, self, LU_data, LU_pivots);
1880 } catch (...) {
1881 }
1882 }
nnc_aten_lgamma(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1883 static void nnc_aten_lgamma(
1884 int64_t bufs_num,
1885 void** buf_data,
1886 int64_t* buf_ranks,
1887 int64_t* buf_dims,
1888 int64_t* buf_strides,
1889 int8_t* buf_dtypes,
1890 int64_t args_num,
1891 int64_t* extra_args) {
1892 std::vector<at::Tensor> tensors = constructTensors(
1893 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1894 at::Tensor& r = tensors[0];
1895 const at::Tensor& self = tensors[1];
1896 try {
1897 at::lgamma_out(r, self);
1898 } catch (...) {
1899 }
1900 }
nnc_aten_digamma(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1901 static void nnc_aten_digamma(
1902 int64_t bufs_num,
1903 void** buf_data,
1904 int64_t* buf_ranks,
1905 int64_t* buf_dims,
1906 int64_t* buf_strides,
1907 int8_t* buf_dtypes,
1908 int64_t args_num,
1909 int64_t* extra_args) {
1910 std::vector<at::Tensor> tensors = constructTensors(
1911 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1912 at::Tensor& r = tensors[0];
1913 const at::Tensor& self = tensors[1];
1914 try {
1915 at::digamma_out(r, self);
1916 } catch (...) {
1917 }
1918 }
nnc_aten_erfinv(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1919 static void nnc_aten_erfinv(
1920 int64_t bufs_num,
1921 void** buf_data,
1922 int64_t* buf_ranks,
1923 int64_t* buf_dims,
1924 int64_t* buf_strides,
1925 int8_t* buf_dtypes,
1926 int64_t args_num,
1927 int64_t* extra_args) {
1928 std::vector<at::Tensor> tensors = constructTensors(
1929 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1930 at::Tensor& r = tensors[0];
1931 const at::Tensor& self = tensors[1];
1932 try {
1933 at::erfinv_out(r, self);
1934 } catch (...) {
1935 }
1936 }
nnc_aten_i0(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1937 static void nnc_aten_i0(
1938 int64_t bufs_num,
1939 void** buf_data,
1940 int64_t* buf_ranks,
1941 int64_t* buf_dims,
1942 int64_t* buf_strides,
1943 int8_t* buf_dtypes,
1944 int64_t args_num,
1945 int64_t* extra_args) {
1946 std::vector<at::Tensor> tensors = constructTensors(
1947 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1948 at::Tensor& r = tensors[0];
1949 const at::Tensor& self = tensors[1];
1950 try {
1951 at::i0_out(r, self);
1952 } catch (...) {
1953 }
1954 }
nnc_aten_sign(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1955 static void nnc_aten_sign(
1956 int64_t bufs_num,
1957 void** buf_data,
1958 int64_t* buf_ranks,
1959 int64_t* buf_dims,
1960 int64_t* buf_strides,
1961 int8_t* buf_dtypes,
1962 int64_t args_num,
1963 int64_t* extra_args) {
1964 std::vector<at::Tensor> tensors = constructTensors(
1965 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1966 at::Tensor& r = tensors[0];
1967 const at::Tensor& self = tensors[1];
1968 try {
1969 at::sign_out(r, self);
1970 } catch (...) {
1971 }
1972 }
nnc_aten_signbit(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1973 static void nnc_aten_signbit(
1974 int64_t bufs_num,
1975 void** buf_data,
1976 int64_t* buf_ranks,
1977 int64_t* buf_dims,
1978 int64_t* buf_strides,
1979 int8_t* buf_dtypes,
1980 int64_t args_num,
1981 int64_t* extra_args) {
1982 std::vector<at::Tensor> tensors = constructTensors(
1983 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
1984 at::Tensor& r = tensors[0];
1985 const at::Tensor& self = tensors[1];
1986 try {
1987 at::signbit_out(r, self);
1988 } catch (...) {
1989 }
1990 }
nnc_aten_atan2(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)1991 static void nnc_aten_atan2(
1992 int64_t bufs_num,
1993 void** buf_data,
1994 int64_t* buf_ranks,
1995 int64_t* buf_dims,
1996 int64_t* buf_strides,
1997 int8_t* buf_dtypes,
1998 int64_t args_num,
1999 int64_t* extra_args) {
2000 std::vector<at::Tensor> tensors = constructTensors(
2001 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2002 at::Tensor& r = tensors[0];
2003 const at::Tensor& self = tensors[1];
2004 const at::Tensor& other = tensors[2];
2005 try {
2006 at::atan2_out(r, self, other);
2007 } catch (...) {
2008 }
2009 }
nnc_aten_lerp(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2010 static void nnc_aten_lerp(
2011 int64_t bufs_num,
2012 void** buf_data,
2013 int64_t* buf_ranks,
2014 int64_t* buf_dims,
2015 int64_t* buf_strides,
2016 int8_t* buf_dtypes,
2017 int64_t args_num,
2018 int64_t* extra_args) {
2019 std::vector<at::Tensor> tensors = constructTensors(
2020 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2021 at::Tensor& r = tensors[0];
2022 const at::Tensor& self = tensors[1];
2023 const at::Tensor& end = tensors[2];
2024 const at::Tensor& weight = tensors[3];
2025 try {
2026 at::lerp_out(r, self, end, weight);
2027 } catch (...) {
2028 }
2029 }
nnc_aten_fmod(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2030 static void nnc_aten_fmod(
2031 int64_t bufs_num,
2032 void** buf_data,
2033 int64_t* buf_ranks,
2034 int64_t* buf_dims,
2035 int64_t* buf_strides,
2036 int8_t* buf_dtypes,
2037 int64_t args_num,
2038 int64_t* extra_args) {
2039 std::vector<at::Tensor> tensors = constructTensors(
2040 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2041 at::Tensor& r = tensors[0];
2042 const at::Tensor& self = tensors[1];
2043 const at::Tensor& other = tensors[2];
2044 try {
2045 at::fmod_out(r, self, other);
2046 } catch (...) {
2047 }
2048 }
nnc_aten_hypot(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2049 static void nnc_aten_hypot(
2050 int64_t bufs_num,
2051 void** buf_data,
2052 int64_t* buf_ranks,
2053 int64_t* buf_dims,
2054 int64_t* buf_strides,
2055 int8_t* buf_dtypes,
2056 int64_t args_num,
2057 int64_t* extra_args) {
2058 std::vector<at::Tensor> tensors = constructTensors(
2059 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2060 at::Tensor& r = tensors[0];
2061 const at::Tensor& self = tensors[1];
2062 const at::Tensor& other = tensors[2];
2063 try {
2064 at::hypot_out(r, self, other);
2065 } catch (...) {
2066 }
2067 }
nnc_aten_igamma(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2068 static void nnc_aten_igamma(
2069 int64_t bufs_num,
2070 void** buf_data,
2071 int64_t* buf_ranks,
2072 int64_t* buf_dims,
2073 int64_t* buf_strides,
2074 int8_t* buf_dtypes,
2075 int64_t args_num,
2076 int64_t* extra_args) {
2077 std::vector<at::Tensor> tensors = constructTensors(
2078 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2079 at::Tensor& r = tensors[0];
2080 const at::Tensor& self = tensors[1];
2081 const at::Tensor& other = tensors[2];
2082 try {
2083 at::igamma_out(r, self, other);
2084 } catch (...) {
2085 }
2086 }
nnc_aten_igammac(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2087 static void nnc_aten_igammac(
2088 int64_t bufs_num,
2089 void** buf_data,
2090 int64_t* buf_ranks,
2091 int64_t* buf_dims,
2092 int64_t* buf_strides,
2093 int8_t* buf_dtypes,
2094 int64_t args_num,
2095 int64_t* extra_args) {
2096 std::vector<at::Tensor> tensors = constructTensors(
2097 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2098 at::Tensor& r = tensors[0];
2099 const at::Tensor& self = tensors[1];
2100 const at::Tensor& other = tensors[2];
2101 try {
2102 at::igammac_out(r, self, other);
2103 } catch (...) {
2104 }
2105 }
nnc_aten_nextafter(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2106 static void nnc_aten_nextafter(
2107 int64_t bufs_num,
2108 void** buf_data,
2109 int64_t* buf_ranks,
2110 int64_t* buf_dims,
2111 int64_t* buf_strides,
2112 int8_t* buf_dtypes,
2113 int64_t args_num,
2114 int64_t* extra_args) {
2115 std::vector<at::Tensor> tensors = constructTensors(
2116 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2117 at::Tensor& r = tensors[0];
2118 const at::Tensor& self = tensors[1];
2119 const at::Tensor& other = tensors[2];
2120 try {
2121 at::nextafter_out(r, self, other);
2122 } catch (...) {
2123 }
2124 }
nnc_aten_remainder(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2125 static void nnc_aten_remainder(
2126 int64_t bufs_num,
2127 void** buf_data,
2128 int64_t* buf_ranks,
2129 int64_t* buf_dims,
2130 int64_t* buf_strides,
2131 int8_t* buf_dtypes,
2132 int64_t args_num,
2133 int64_t* extra_args) {
2134 std::vector<at::Tensor> tensors = constructTensors(
2135 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2136 at::Tensor& r = tensors[0];
2137 const at::Tensor& self = tensors[1];
2138 const at::Tensor& other = tensors[2];
2139 try {
2140 at::remainder_out(r, self, other);
2141 } catch (...) {
2142 }
2143 }
nnc_aten_fmin(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2144 static void nnc_aten_fmin(
2145 int64_t bufs_num,
2146 void** buf_data,
2147 int64_t* buf_ranks,
2148 int64_t* buf_dims,
2149 int64_t* buf_strides,
2150 int8_t* buf_dtypes,
2151 int64_t args_num,
2152 int64_t* extra_args) {
2153 std::vector<at::Tensor> tensors = constructTensors(
2154 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2155 at::Tensor& r = tensors[0];
2156 const at::Tensor& self = tensors[1];
2157 const at::Tensor& other = tensors[2];
2158 try {
2159 at::fmin_out(r, self, other);
2160 } catch (...) {
2161 }
2162 }
nnc_aten_fmax(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2163 static void nnc_aten_fmax(
2164 int64_t bufs_num,
2165 void** buf_data,
2166 int64_t* buf_ranks,
2167 int64_t* buf_dims,
2168 int64_t* buf_strides,
2169 int8_t* buf_dtypes,
2170 int64_t args_num,
2171 int64_t* extra_args) {
2172 std::vector<at::Tensor> tensors = constructTensors(
2173 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2174 at::Tensor& r = tensors[0];
2175 const at::Tensor& self = tensors[1];
2176 const at::Tensor& other = tensors[2];
2177 try {
2178 at::fmax_out(r, self, other);
2179 } catch (...) {
2180 }
2181 }
nnc_aten_maximum(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2182 static void nnc_aten_maximum(
2183 int64_t bufs_num,
2184 void** buf_data,
2185 int64_t* buf_ranks,
2186 int64_t* buf_dims,
2187 int64_t* buf_strides,
2188 int8_t* buf_dtypes,
2189 int64_t args_num,
2190 int64_t* extra_args) {
2191 std::vector<at::Tensor> tensors = constructTensors(
2192 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2193 at::Tensor& r = tensors[0];
2194 const at::Tensor& self = tensors[1];
2195 const at::Tensor& other = tensors[2];
2196 try {
2197 at::maximum_out(r, self, other);
2198 } catch (...) {
2199 }
2200 }
nnc_aten_max(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2201 static void nnc_aten_max(
2202 int64_t bufs_num,
2203 void** buf_data,
2204 int64_t* buf_ranks,
2205 int64_t* buf_dims,
2206 int64_t* buf_strides,
2207 int8_t* buf_dtypes,
2208 int64_t args_num,
2209 int64_t* extra_args) {
2210 std::vector<at::Tensor> tensors = constructTensors(
2211 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2212 at::Tensor& r = tensors[0];
2213 const at::Tensor& self = tensors[1];
2214 const at::Tensor& other = tensors[2];
2215 try {
2216 at::max_out(r, self, other);
2217 } catch (...) {
2218 }
2219 }
nnc_aten_minimum(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2220 static void nnc_aten_minimum(
2221 int64_t bufs_num,
2222 void** buf_data,
2223 int64_t* buf_ranks,
2224 int64_t* buf_dims,
2225 int64_t* buf_strides,
2226 int8_t* buf_dtypes,
2227 int64_t args_num,
2228 int64_t* extra_args) {
2229 std::vector<at::Tensor> tensors = constructTensors(
2230 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2231 at::Tensor& r = tensors[0];
2232 const at::Tensor& self = tensors[1];
2233 const at::Tensor& other = tensors[2];
2234 try {
2235 at::minimum_out(r, self, other);
2236 } catch (...) {
2237 }
2238 }
nnc_aten_min(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2239 static void nnc_aten_min(
2240 int64_t bufs_num,
2241 void** buf_data,
2242 int64_t* buf_ranks,
2243 int64_t* buf_dims,
2244 int64_t* buf_strides,
2245 int8_t* buf_dtypes,
2246 int64_t args_num,
2247 int64_t* extra_args) {
2248 std::vector<at::Tensor> tensors = constructTensors(
2249 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2250 at::Tensor& r = tensors[0];
2251 const at::Tensor& self = tensors[1];
2252 const at::Tensor& other = tensors[2];
2253 try {
2254 at::min_out(r, self, other);
2255 } catch (...) {
2256 }
2257 }
nnc_aten_msort(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2258 static void nnc_aten_msort(
2259 int64_t bufs_num,
2260 void** buf_data,
2261 int64_t* buf_ranks,
2262 int64_t* buf_dims,
2263 int64_t* buf_strides,
2264 int8_t* buf_dtypes,
2265 int64_t args_num,
2266 int64_t* extra_args) {
2267 std::vector<at::Tensor> tensors = constructTensors(
2268 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2269 at::Tensor& r = tensors[0];
2270 const at::Tensor& self = tensors[1];
2271 try {
2272 at::msort_out(r, self);
2273 } catch (...) {
2274 }
2275 }
nnc_aten_pow(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2276 static void nnc_aten_pow(
2277 int64_t bufs_num,
2278 void** buf_data,
2279 int64_t* buf_ranks,
2280 int64_t* buf_dims,
2281 int64_t* buf_strides,
2282 int8_t* buf_dtypes,
2283 int64_t args_num,
2284 int64_t* extra_args) {
2285 std::vector<at::Tensor> tensors = constructTensors(
2286 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2287 at::Tensor& r = tensors[0];
2288 const at::Tensor& self = tensors[1];
2289 const at::Tensor& exponent = tensors[2];
2290 try {
2291 at::pow_out(r, self, exponent);
2292 } catch (...) {
2293 }
2294 }
nnc_aten_float_power(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2295 static void nnc_aten_float_power(
2296 int64_t bufs_num,
2297 void** buf_data,
2298 int64_t* buf_ranks,
2299 int64_t* buf_dims,
2300 int64_t* buf_strides,
2301 int8_t* buf_dtypes,
2302 int64_t args_num,
2303 int64_t* extra_args) {
2304 std::vector<at::Tensor> tensors = constructTensors(
2305 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2306 at::Tensor& r = tensors[0];
2307 const at::Tensor& self = tensors[1];
2308 const at::Tensor& exponent = tensors[2];
2309 try {
2310 at::float_power_out(r, self, exponent);
2311 } catch (...) {
2312 }
2313 }
nnc_aten_hardsigmoid(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2314 static void nnc_aten_hardsigmoid(
2315 int64_t bufs_num,
2316 void** buf_data,
2317 int64_t* buf_ranks,
2318 int64_t* buf_dims,
2319 int64_t* buf_strides,
2320 int8_t* buf_dtypes,
2321 int64_t args_num,
2322 int64_t* extra_args) {
2323 std::vector<at::Tensor> tensors = constructTensors(
2324 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2325 at::Tensor& r = tensors[0];
2326 const at::Tensor& self = tensors[1];
2327 try {
2328 at::hardsigmoid_out(r, self);
2329 } catch (...) {
2330 }
2331 }
nnc_aten_hardswish(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2332 static void nnc_aten_hardswish(
2333 int64_t bufs_num,
2334 void** buf_data,
2335 int64_t* buf_ranks,
2336 int64_t* buf_dims,
2337 int64_t* buf_strides,
2338 int8_t* buf_dtypes,
2339 int64_t args_num,
2340 int64_t* extra_args) {
2341 std::vector<at::Tensor> tensors = constructTensors(
2342 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2343 at::Tensor& r = tensors[0];
2344 const at::Tensor& self = tensors[1];
2345 try {
2346 at::hardswish_out(r, self);
2347 } catch (...) {
2348 }
2349 }
nnc_aten_log_sigmoid(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2350 static void nnc_aten_log_sigmoid(
2351 int64_t bufs_num,
2352 void** buf_data,
2353 int64_t* buf_ranks,
2354 int64_t* buf_dims,
2355 int64_t* buf_strides,
2356 int8_t* buf_dtypes,
2357 int64_t args_num,
2358 int64_t* extra_args) {
2359 std::vector<at::Tensor> tensors = constructTensors(
2360 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2361 at::Tensor& r = tensors[0];
2362 const at::Tensor& self = tensors[1];
2363 try {
2364 at::log_sigmoid_out(r, self);
2365 } catch (...) {
2366 }
2367 }
nnc_aten_log_sigmoid_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2368 static void nnc_aten_log_sigmoid_backward(
2369 int64_t bufs_num,
2370 void** buf_data,
2371 int64_t* buf_ranks,
2372 int64_t* buf_dims,
2373 int64_t* buf_strides,
2374 int8_t* buf_dtypes,
2375 int64_t args_num,
2376 int64_t* extra_args) {
2377 std::vector<at::Tensor> tensors = constructTensors(
2378 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2379 at::Tensor& r = tensors[0];
2380 const at::Tensor& grad_output = tensors[1];
2381 const at::Tensor& self = tensors[2];
2382 const at::Tensor& buffer = tensors[3];
2383 try {
2384 at::log_sigmoid_backward_out(r, grad_output, self, buffer);
2385 } catch (...) {
2386 }
2387 }
nnc_aten_adaptive_avg_pool3d_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2388 static void nnc_aten_adaptive_avg_pool3d_backward(
2389 int64_t bufs_num,
2390 void** buf_data,
2391 int64_t* buf_ranks,
2392 int64_t* buf_dims,
2393 int64_t* buf_strides,
2394 int8_t* buf_dtypes,
2395 int64_t args_num,
2396 int64_t* extra_args) {
2397 std::vector<at::Tensor> tensors = constructTensors(
2398 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2399 at::Tensor& r = tensors[0];
2400 const at::Tensor& grad_output = tensors[1];
2401 const at::Tensor& self = tensors[2];
2402 try {
2403 at::adaptive_avg_pool3d_backward_out(r, grad_output, self);
2404 } catch (...) {
2405 }
2406 }
nnc_aten_adaptive_max_pool2d_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2407 static void nnc_aten_adaptive_max_pool2d_backward(
2408 int64_t bufs_num,
2409 void** buf_data,
2410 int64_t* buf_ranks,
2411 int64_t* buf_dims,
2412 int64_t* buf_strides,
2413 int8_t* buf_dtypes,
2414 int64_t args_num,
2415 int64_t* extra_args) {
2416 std::vector<at::Tensor> tensors = constructTensors(
2417 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2418 at::Tensor& r = tensors[0];
2419 const at::Tensor& grad_output = tensors[1];
2420 const at::Tensor& self = tensors[2];
2421 const at::Tensor& indices = tensors[3];
2422 try {
2423 at::adaptive_max_pool2d_backward_out(r, grad_output, self, indices);
2424 } catch (...) {
2425 }
2426 }
nnc_aten_adaptive_max_pool3d_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2427 static void nnc_aten_adaptive_max_pool3d_backward(
2428 int64_t bufs_num,
2429 void** buf_data,
2430 int64_t* buf_ranks,
2431 int64_t* buf_dims,
2432 int64_t* buf_strides,
2433 int8_t* buf_dtypes,
2434 int64_t args_num,
2435 int64_t* extra_args) {
2436 std::vector<at::Tensor> tensors = constructTensors(
2437 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2438 at::Tensor& r = tensors[0];
2439 const at::Tensor& grad_output = tensors[1];
2440 const at::Tensor& self = tensors[2];
2441 const at::Tensor& indices = tensors[3];
2442 try {
2443 at::adaptive_max_pool3d_backward_out(r, grad_output, self, indices);
2444 } catch (...) {
2445 }
2446 }
nnc_aten_sigmoid_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2447 static void nnc_aten_sigmoid_backward(
2448 int64_t bufs_num,
2449 void** buf_data,
2450 int64_t* buf_ranks,
2451 int64_t* buf_dims,
2452 int64_t* buf_strides,
2453 int8_t* buf_dtypes,
2454 int64_t args_num,
2455 int64_t* extra_args) {
2456 std::vector<at::Tensor> tensors = constructTensors(
2457 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2458 at::Tensor& r = tensors[0];
2459 const at::Tensor& grad_output = tensors[1];
2460 const at::Tensor& output = tensors[2];
2461 try {
2462 at::sigmoid_backward_out(r, grad_output, output);
2463 } catch (...) {
2464 }
2465 }
nnc_aten_tanh_backward(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2466 static void nnc_aten_tanh_backward(
2467 int64_t bufs_num,
2468 void** buf_data,
2469 int64_t* buf_ranks,
2470 int64_t* buf_dims,
2471 int64_t* buf_strides,
2472 int8_t* buf_dtypes,
2473 int64_t args_num,
2474 int64_t* extra_args) {
2475 std::vector<at::Tensor> tensors = constructTensors(
2476 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2477 at::Tensor& r = tensors[0];
2478 const at::Tensor& grad_output = tensors[1];
2479 const at::Tensor& output = tensors[2];
2480 try {
2481 at::tanh_backward_out(r, grad_output, output);
2482 } catch (...) {
2483 }
2484 }
nnc_aten_isposinf(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2485 static void nnc_aten_isposinf(
2486 int64_t bufs_num,
2487 void** buf_data,
2488 int64_t* buf_ranks,
2489 int64_t* buf_dims,
2490 int64_t* buf_strides,
2491 int8_t* buf_dtypes,
2492 int64_t args_num,
2493 int64_t* extra_args) {
2494 std::vector<at::Tensor> tensors = constructTensors(
2495 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2496 at::Tensor& r = tensors[0];
2497 const at::Tensor& self = tensors[1];
2498 try {
2499 at::isposinf_out(r, self);
2500 } catch (...) {
2501 }
2502 }
nnc_aten_isneginf(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2503 static void nnc_aten_isneginf(
2504 int64_t bufs_num,
2505 void** buf_data,
2506 int64_t* buf_ranks,
2507 int64_t* buf_dims,
2508 int64_t* buf_strides,
2509 int8_t* buf_dtypes,
2510 int64_t args_num,
2511 int64_t* extra_args) {
2512 std::vector<at::Tensor> tensors = constructTensors(
2513 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2514 at::Tensor& r = tensors[0];
2515 const at::Tensor& self = tensors[1];
2516 try {
2517 at::isneginf_out(r, self);
2518 } catch (...) {
2519 }
2520 }
nnc_aten_special_entr(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2521 static void nnc_aten_special_entr(
2522 int64_t bufs_num,
2523 void** buf_data,
2524 int64_t* buf_ranks,
2525 int64_t* buf_dims,
2526 int64_t* buf_strides,
2527 int8_t* buf_dtypes,
2528 int64_t args_num,
2529 int64_t* extra_args) {
2530 std::vector<at::Tensor> tensors = constructTensors(
2531 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2532 at::Tensor& r = tensors[0];
2533 const at::Tensor& self = tensors[1];
2534 try {
2535 at::special_entr_out(r, self);
2536 } catch (...) {
2537 }
2538 }
nnc_aten_special_expm1(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2539 static void nnc_aten_special_expm1(
2540 int64_t bufs_num,
2541 void** buf_data,
2542 int64_t* buf_ranks,
2543 int64_t* buf_dims,
2544 int64_t* buf_strides,
2545 int8_t* buf_dtypes,
2546 int64_t args_num,
2547 int64_t* extra_args) {
2548 std::vector<at::Tensor> tensors = constructTensors(
2549 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2550 at::Tensor& r = tensors[0];
2551 const at::Tensor& self = tensors[1];
2552 try {
2553 at::special_expm1_out(r, self);
2554 } catch (...) {
2555 }
2556 }
nnc_aten_special_exp2(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2557 static void nnc_aten_special_exp2(
2558 int64_t bufs_num,
2559 void** buf_data,
2560 int64_t* buf_ranks,
2561 int64_t* buf_dims,
2562 int64_t* buf_strides,
2563 int8_t* buf_dtypes,
2564 int64_t args_num,
2565 int64_t* extra_args) {
2566 std::vector<at::Tensor> tensors = constructTensors(
2567 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2568 at::Tensor& r = tensors[0];
2569 const at::Tensor& self = tensors[1];
2570 try {
2571 at::special_exp2_out(r, self);
2572 } catch (...) {
2573 }
2574 }
nnc_aten_special_gammaln(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2575 static void nnc_aten_special_gammaln(
2576 int64_t bufs_num,
2577 void** buf_data,
2578 int64_t* buf_ranks,
2579 int64_t* buf_dims,
2580 int64_t* buf_strides,
2581 int8_t* buf_dtypes,
2582 int64_t args_num,
2583 int64_t* extra_args) {
2584 std::vector<at::Tensor> tensors = constructTensors(
2585 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2586 at::Tensor& r = tensors[0];
2587 const at::Tensor& self = tensors[1];
2588 try {
2589 at::special_gammaln_out(r, self);
2590 } catch (...) {
2591 }
2592 }
nnc_aten_special_erf(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2593 static void nnc_aten_special_erf(
2594 int64_t bufs_num,
2595 void** buf_data,
2596 int64_t* buf_ranks,
2597 int64_t* buf_dims,
2598 int64_t* buf_strides,
2599 int8_t* buf_dtypes,
2600 int64_t args_num,
2601 int64_t* extra_args) {
2602 std::vector<at::Tensor> tensors = constructTensors(
2603 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2604 at::Tensor& r = tensors[0];
2605 const at::Tensor& self = tensors[1];
2606 try {
2607 at::special_erf_out(r, self);
2608 } catch (...) {
2609 }
2610 }
nnc_aten_special_erfc(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2611 static void nnc_aten_special_erfc(
2612 int64_t bufs_num,
2613 void** buf_data,
2614 int64_t* buf_ranks,
2615 int64_t* buf_dims,
2616 int64_t* buf_strides,
2617 int8_t* buf_dtypes,
2618 int64_t args_num,
2619 int64_t* extra_args) {
2620 std::vector<at::Tensor> tensors = constructTensors(
2621 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2622 at::Tensor& r = tensors[0];
2623 const at::Tensor& self = tensors[1];
2624 try {
2625 at::special_erfc_out(r, self);
2626 } catch (...) {
2627 }
2628 }
nnc_aten_special_erfinv(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2629 static void nnc_aten_special_erfinv(
2630 int64_t bufs_num,
2631 void** buf_data,
2632 int64_t* buf_ranks,
2633 int64_t* buf_dims,
2634 int64_t* buf_strides,
2635 int8_t* buf_dtypes,
2636 int64_t args_num,
2637 int64_t* extra_args) {
2638 std::vector<at::Tensor> tensors = constructTensors(
2639 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2640 at::Tensor& r = tensors[0];
2641 const at::Tensor& self = tensors[1];
2642 try {
2643 at::special_erfinv_out(r, self);
2644 } catch (...) {
2645 }
2646 }
nnc_aten_special_xlog1py(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2647 static void nnc_aten_special_xlog1py(
2648 int64_t bufs_num,
2649 void** buf_data,
2650 int64_t* buf_ranks,
2651 int64_t* buf_dims,
2652 int64_t* buf_strides,
2653 int8_t* buf_dtypes,
2654 int64_t args_num,
2655 int64_t* extra_args) {
2656 std::vector<at::Tensor> tensors = constructTensors(
2657 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2658 at::Tensor& r = tensors[0];
2659 const at::Tensor& self = tensors[1];
2660 const at::Tensor& other = tensors[2];
2661 try {
2662 at::special_xlog1py_out(r, self, other);
2663 } catch (...) {
2664 }
2665 }
nnc_aten_special_i0e(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2666 static void nnc_aten_special_i0e(
2667 int64_t bufs_num,
2668 void** buf_data,
2669 int64_t* buf_ranks,
2670 int64_t* buf_dims,
2671 int64_t* buf_strides,
2672 int8_t* buf_dtypes,
2673 int64_t args_num,
2674 int64_t* extra_args) {
2675 std::vector<at::Tensor> tensors = constructTensors(
2676 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2677 at::Tensor& r = tensors[0];
2678 const at::Tensor& self = tensors[1];
2679 try {
2680 at::special_i0e_out(r, self);
2681 } catch (...) {
2682 }
2683 }
nnc_aten_special_expit(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2684 static void nnc_aten_special_expit(
2685 int64_t bufs_num,
2686 void** buf_data,
2687 int64_t* buf_ranks,
2688 int64_t* buf_dims,
2689 int64_t* buf_strides,
2690 int8_t* buf_dtypes,
2691 int64_t args_num,
2692 int64_t* extra_args) {
2693 std::vector<at::Tensor> tensors = constructTensors(
2694 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2695 at::Tensor& r = tensors[0];
2696 const at::Tensor& self = tensors[1];
2697 try {
2698 at::special_expit_out(r, self);
2699 } catch (...) {
2700 }
2701 }
nnc_aten_linalg_cholesky(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2702 static void nnc_aten_linalg_cholesky(
2703 int64_t bufs_num,
2704 void** buf_data,
2705 int64_t* buf_ranks,
2706 int64_t* buf_dims,
2707 int64_t* buf_strides,
2708 int8_t* buf_dtypes,
2709 int64_t args_num,
2710 int64_t* extra_args) {
2711 std::vector<at::Tensor> tensors = constructTensors(
2712 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2713 at::Tensor& r = tensors[0];
2714 const at::Tensor& self = tensors[1];
2715 try {
2716 at::linalg_cholesky_out(r, self);
2717 } catch (...) {
2718 }
2719 }
nnc_aten_linalg_det(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2720 static void nnc_aten_linalg_det(
2721 int64_t bufs_num,
2722 void** buf_data,
2723 int64_t* buf_ranks,
2724 int64_t* buf_dims,
2725 int64_t* buf_strides,
2726 int8_t* buf_dtypes,
2727 int64_t args_num,
2728 int64_t* extra_args) {
2729 std::vector<at::Tensor> tensors = constructTensors(
2730 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2731 at::Tensor& r = tensors[0];
2732 const at::Tensor& self = tensors[1];
2733 try {
2734 at::linalg_det_out(r, self);
2735 } catch (...) {
2736 }
2737 }
nnc_aten_linalg_eigvals(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2738 static void nnc_aten_linalg_eigvals(
2739 int64_t bufs_num,
2740 void** buf_data,
2741 int64_t* buf_ranks,
2742 int64_t* buf_dims,
2743 int64_t* buf_strides,
2744 int8_t* buf_dtypes,
2745 int64_t args_num,
2746 int64_t* extra_args) {
2747 std::vector<at::Tensor> tensors = constructTensors(
2748 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2749 at::Tensor& r = tensors[0];
2750 const at::Tensor& self = tensors[1];
2751 try {
2752 at::linalg_eigvals_out(r, self);
2753 } catch (...) {
2754 }
2755 }
nnc_aten_linalg_householder_product(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2756 static void nnc_aten_linalg_householder_product(
2757 int64_t bufs_num,
2758 void** buf_data,
2759 int64_t* buf_ranks,
2760 int64_t* buf_dims,
2761 int64_t* buf_strides,
2762 int8_t* buf_dtypes,
2763 int64_t args_num,
2764 int64_t* extra_args) {
2765 std::vector<at::Tensor> tensors = constructTensors(
2766 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2767 at::Tensor& r = tensors[0];
2768 const at::Tensor& input = tensors[1];
2769 const at::Tensor& tau = tensors[2];
2770 try {
2771 at::linalg_householder_product_out(r, input, tau);
2772 } catch (...) {
2773 }
2774 }
nnc_aten_linalg_inv(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2775 static void nnc_aten_linalg_inv(
2776 int64_t bufs_num,
2777 void** buf_data,
2778 int64_t* buf_ranks,
2779 int64_t* buf_dims,
2780 int64_t* buf_strides,
2781 int8_t* buf_dtypes,
2782 int64_t args_num,
2783 int64_t* extra_args) {
2784 std::vector<at::Tensor> tensors = constructTensors(
2785 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2786 at::Tensor& r = tensors[0];
2787 const at::Tensor& self = tensors[1];
2788 try {
2789 at::linalg_inv_out(r, self);
2790 } catch (...) {
2791 }
2792 }
nnc_aten_inner(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2793 static void nnc_aten_inner(
2794 int64_t bufs_num,
2795 void** buf_data,
2796 int64_t* buf_ranks,
2797 int64_t* buf_dims,
2798 int64_t* buf_strides,
2799 int8_t* buf_dtypes,
2800 int64_t args_num,
2801 int64_t* extra_args) {
2802 std::vector<at::Tensor> tensors = constructTensors(
2803 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2804 at::Tensor& r = tensors[0];
2805 const at::Tensor& self = tensors[1];
2806 const at::Tensor& other = tensors[2];
2807 try {
2808 at::inner_out(r, self, other);
2809 } catch (...) {
2810 }
2811 }
nnc_aten_outer(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2812 static void nnc_aten_outer(
2813 int64_t bufs_num,
2814 void** buf_data,
2815 int64_t* buf_ranks,
2816 int64_t* buf_dims,
2817 int64_t* buf_strides,
2818 int8_t* buf_dtypes,
2819 int64_t args_num,
2820 int64_t* extra_args) {
2821 std::vector<at::Tensor> tensors = constructTensors(
2822 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2823 at::Tensor& r = tensors[0];
2824 const at::Tensor& self = tensors[1];
2825 const at::Tensor& vec2 = tensors[2];
2826 try {
2827 at::outer_out(r, self, vec2);
2828 } catch (...) {
2829 }
2830 }
nnc_aten_ger(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2831 static void nnc_aten_ger(
2832 int64_t bufs_num,
2833 void** buf_data,
2834 int64_t* buf_ranks,
2835 int64_t* buf_dims,
2836 int64_t* buf_strides,
2837 int8_t* buf_dtypes,
2838 int64_t args_num,
2839 int64_t* extra_args) {
2840 std::vector<at::Tensor> tensors = constructTensors(
2841 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2842 at::Tensor& r = tensors[0];
2843 const at::Tensor& self = tensors[1];
2844 const at::Tensor& vec2 = tensors[2];
2845 try {
2846 at::ger_out(r, self, vec2);
2847 } catch (...) {
2848 }
2849 }
nnc_aten_linalg_svdvals(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2850 static void nnc_aten_linalg_svdvals(
2851 int64_t bufs_num,
2852 void** buf_data,
2853 int64_t* buf_ranks,
2854 int64_t* buf_dims,
2855 int64_t* buf_strides,
2856 int8_t* buf_dtypes,
2857 int64_t args_num,
2858 int64_t* extra_args) {
2859 std::vector<at::Tensor> tensors = constructTensors(
2860 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2861 at::Tensor& r = tensors[0];
2862 const at::Tensor& input = tensors[1];
2863 try {
2864 at::linalg_svdvals_out(r, input);
2865 } catch (...) {
2866 }
2867 }
nnc_aten_linalg_solve(int64_t bufs_num,void ** buf_data,int64_t * buf_ranks,int64_t * buf_dims,int64_t * buf_strides,int8_t * buf_dtypes,int64_t args_num,int64_t * extra_args)2868 static void nnc_aten_linalg_solve(
2869 int64_t bufs_num,
2870 void** buf_data,
2871 int64_t* buf_ranks,
2872 int64_t* buf_dims,
2873 int64_t* buf_strides,
2874 int8_t* buf_dtypes,
2875 int64_t args_num,
2876 int64_t* extra_args) {
2877 std::vector<at::Tensor> tensors = constructTensors(
2878 bufs_num, buf_data, buf_ranks, buf_dims, buf_strides, buf_dtypes);
2879 at::Tensor& r = tensors[0];
2880 const at::Tensor& input = tensors[1];
2881 const at::Tensor& other = tensors[2];
2882 try {
2883 at::linalg_solve_out(r, input, other, true);
2884 } catch (...) {
2885 }
2886 }
2887
2888 const static RegisterNNCExternalFunction nnc_abs("nnc_aten_abs", nnc_aten_abs);
2889 const static RegisterNNCExternalFunction nnc_absolute(
2890 "nnc_aten_absolute",
2891 nnc_aten_absolute);
2892 const static RegisterNNCExternalFunction nnc_angle(
2893 "nnc_aten_angle",
2894 nnc_aten_angle);
2895 const static RegisterNNCExternalFunction nnc_sgn("nnc_aten_sgn", nnc_aten_sgn);
2896 const static RegisterNNCExternalFunction nnc_acos(
2897 "nnc_aten_acos",
2898 nnc_aten_acos);
2899 const static RegisterNNCExternalFunction nnc_arccos(
2900 "nnc_aten_arccos",
2901 nnc_aten_arccos);
2902 const static RegisterNNCExternalFunction nnc_acosh(
2903 "nnc_aten_acosh",
2904 nnc_aten_acosh);
2905 const static RegisterNNCExternalFunction nnc_arccosh(
2906 "nnc_aten_arccosh",
2907 nnc_aten_arccosh);
2908 const static RegisterNNCExternalFunction nnc_asinh(
2909 "nnc_aten_asinh",
2910 nnc_aten_asinh);
2911 const static RegisterNNCExternalFunction nnc_arcsinh(
2912 "nnc_aten_arcsinh",
2913 nnc_aten_arcsinh);
2914 const static RegisterNNCExternalFunction nnc_atanh(
2915 "nnc_aten_atanh",
2916 nnc_aten_atanh);
2917 const static RegisterNNCExternalFunction nnc_arctanh(
2918 "nnc_aten_arctanh",
2919 nnc_aten_arctanh);
2920 const static RegisterNNCExternalFunction nnc_asin(
2921 "nnc_aten_asin",
2922 nnc_aten_asin);
2923 const static RegisterNNCExternalFunction nnc_arcsin(
2924 "nnc_aten_arcsin",
2925 nnc_aten_arcsin);
2926 const static RegisterNNCExternalFunction nnc_atan(
2927 "nnc_aten_atan",
2928 nnc_aten_atan);
2929 const static RegisterNNCExternalFunction nnc_arctan(
2930 "nnc_aten_arctan",
2931 nnc_aten_arctan);
2932 const static RegisterNNCExternalFunction nnc_bitwise_not(
2933 "nnc_aten_bitwise_not",
2934 nnc_aten_bitwise_not);
2935 const static RegisterNNCExternalFunction nnc_copysign(
2936 "nnc_aten_copysign",
2937 nnc_aten_copysign);
2938 const static RegisterNNCExternalFunction nnc_logical_not(
2939 "nnc_aten_logical_not",
2940 nnc_aten_logical_not);
2941 const static RegisterNNCExternalFunction nnc_logical_xor(
2942 "nnc_aten_logical_xor",
2943 nnc_aten_logical_xor);
2944 const static RegisterNNCExternalFunction nnc_logical_and(
2945 "nnc_aten_logical_and",
2946 nnc_aten_logical_and);
2947 const static RegisterNNCExternalFunction nnc_logical_or(
2948 "nnc_aten_logical_or",
2949 nnc_aten_logical_or);
2950 const static RegisterNNCExternalFunction nnc_bmm("nnc_aten_bmm", nnc_aten_bmm);
2951 const static RegisterNNCExternalFunction nnc_ceil(
2952 "nnc_aten_ceil",
2953 nnc_aten_ceil);
2954 const static RegisterNNCExternalFunction nnc_clamp_max(
2955 "nnc_aten_clamp_max",
2956 nnc_aten_clamp_max);
2957 const static RegisterNNCExternalFunction nnc_clamp_min(
2958 "nnc_aten_clamp_min",
2959 nnc_aten_clamp_min);
2960 const static RegisterNNCExternalFunction nnc_complex(
2961 "nnc_aten_complex",
2962 nnc_aten_complex);
2963 const static RegisterNNCExternalFunction nnc_polar(
2964 "nnc_aten_polar",
2965 nnc_aten_polar);
2966 const static RegisterNNCExternalFunction nnc_cos("nnc_aten_cos", nnc_aten_cos);
2967 const static RegisterNNCExternalFunction nnc_cosh(
2968 "nnc_aten_cosh",
2969 nnc_aten_cosh);
2970 const static RegisterNNCExternalFunction nnc_div("nnc_aten_div", nnc_aten_div);
2971 const static RegisterNNCExternalFunction nnc_divide(
2972 "nnc_aten_divide",
2973 nnc_aten_divide);
2974 const static RegisterNNCExternalFunction nnc_true_divide(
2975 "nnc_aten_true_divide",
2976 nnc_aten_true_divide);
2977 const static RegisterNNCExternalFunction nnc_dot("nnc_aten_dot", nnc_aten_dot);
2978 const static RegisterNNCExternalFunction nnc_vdot(
2979 "nnc_aten_vdot",
2980 nnc_aten_vdot);
2981 const static RegisterNNCExternalFunction nnc_erf("nnc_aten_erf", nnc_aten_erf);
2982 const static RegisterNNCExternalFunction nnc_erfc(
2983 "nnc_aten_erfc",
2984 nnc_aten_erfc);
2985 const static RegisterNNCExternalFunction nnc_exp("nnc_aten_exp", nnc_aten_exp);
2986 const static RegisterNNCExternalFunction nnc_exp2(
2987 "nnc_aten_exp2",
2988 nnc_aten_exp2);
2989 const static RegisterNNCExternalFunction nnc_expm1(
2990 "nnc_aten_expm1",
2991 nnc_aten_expm1);
2992 const static RegisterNNCExternalFunction nnc_floor(
2993 "nnc_aten_floor",
2994 nnc_aten_floor);
2995 const static RegisterNNCExternalFunction nnc_floor_divide(
2996 "nnc_aten_floor_divide",
2997 nnc_aten_floor_divide);
2998 const static RegisterNNCExternalFunction nnc_frac(
2999 "nnc_aten_frac",
3000 nnc_aten_frac);
3001 const static RegisterNNCExternalFunction nnc_gcd("nnc_aten_gcd", nnc_aten_gcd);
3002 const static RegisterNNCExternalFunction nnc_lcm("nnc_aten_lcm", nnc_aten_lcm);
3003 const static RegisterNNCExternalFunction nnc_inverse(
3004 "nnc_aten_inverse",
3005 nnc_aten_inverse);
3006 const static RegisterNNCExternalFunction nnc_kron(
3007 "nnc_aten_kron",
3008 nnc_aten_kron);
3009 const static RegisterNNCExternalFunction nnc_ldexp(
3010 "nnc_aten_ldexp",
3011 nnc_aten_ldexp);
3012 const static RegisterNNCExternalFunction nnc_log("nnc_aten_log", nnc_aten_log);
3013 const static RegisterNNCExternalFunction nnc_log10(
3014 "nnc_aten_log10",
3015 nnc_aten_log10);
3016 const static RegisterNNCExternalFunction nnc_log1p(
3017 "nnc_aten_log1p",
3018 nnc_aten_log1p);
3019 const static RegisterNNCExternalFunction nnc_log2(
3020 "nnc_aten_log2",
3021 nnc_aten_log2);
3022 const static RegisterNNCExternalFunction nnc_logaddexp(
3023 "nnc_aten_logaddexp",
3024 nnc_aten_logaddexp);
3025 const static RegisterNNCExternalFunction nnc_logaddexp2(
3026 "nnc_aten_logaddexp2",
3027 nnc_aten_logaddexp2);
3028 const static RegisterNNCExternalFunction nnc_xlogy(
3029 "nnc_aten_xlogy",
3030 nnc_aten_xlogy);
3031 const static RegisterNNCExternalFunction nnc_matmul(
3032 "nnc_aten_matmul",
3033 nnc_aten_matmul);
3034 const static RegisterNNCExternalFunction nnc__compute_linear_combination(
3035 "nnc_aten__compute_linear_combination",
3036 nnc_aten__compute_linear_combination);
3037 const static RegisterNNCExternalFunction nnc_mm("nnc_aten_mm", nnc_aten_mm);
3038 const static RegisterNNCExternalFunction nnc_mul("nnc_aten_mul", nnc_aten_mul);
3039 const static RegisterNNCExternalFunction nnc_multiply(
3040 "nnc_aten_multiply",
3041 nnc_aten_multiply);
3042 const static RegisterNNCExternalFunction nnc_mv("nnc_aten_mv", nnc_aten_mv);
3043 const static RegisterNNCExternalFunction nnc_rad2deg(
3044 "nnc_aten_rad2deg",
3045 nnc_aten_rad2deg);
3046 const static RegisterNNCExternalFunction nnc_deg2rad(
3047 "nnc_aten_deg2rad",
3048 nnc_aten_deg2rad);
3049 const static RegisterNNCExternalFunction nnc_reciprocal(
3050 "nnc_aten_reciprocal",
3051 nnc_aten_reciprocal);
3052 const static RegisterNNCExternalFunction nnc_neg("nnc_aten_neg", nnc_aten_neg);
3053 const static RegisterNNCExternalFunction nnc_negative(
3054 "nnc_aten_negative",
3055 nnc_aten_negative);
3056 const static RegisterNNCExternalFunction nnc_round(
3057 "nnc_aten_round",
3058 nnc_aten_round);
3059 const static RegisterNNCExternalFunction nnc_rsqrt(
3060 "nnc_aten_rsqrt",
3061 nnc_aten_rsqrt);
3062 const static RegisterNNCExternalFunction nnc_silu(
3063 "nnc_aten_silu",
3064 nnc_aten_silu);
3065 const static RegisterNNCExternalFunction nnc_mish(
3066 "nnc_aten_mish",
3067 nnc_aten_mish);
3068 const static RegisterNNCExternalFunction nnc_sigmoid(
3069 "nnc_aten_sigmoid",
3070 nnc_aten_sigmoid);
3071 const static RegisterNNCExternalFunction nnc_sin("nnc_aten_sin", nnc_aten_sin);
3072 const static RegisterNNCExternalFunction nnc_sinc(
3073 "nnc_aten_sinc",
3074 nnc_aten_sinc);
3075 const static RegisterNNCExternalFunction nnc_sinh(
3076 "nnc_aten_sinh",
3077 nnc_aten_sinh);
3078 const static RegisterNNCExternalFunction nnc_sqrt(
3079 "nnc_aten_sqrt",
3080 nnc_aten_sqrt);
3081 const static RegisterNNCExternalFunction nnc_square(
3082 "nnc_aten_square",
3083 nnc_aten_square);
3084 const static RegisterNNCExternalFunction nnc_tan("nnc_aten_tan", nnc_aten_tan);
3085 const static RegisterNNCExternalFunction nnc_tanh(
3086 "nnc_aten_tanh",
3087 nnc_aten_tanh);
3088 const static RegisterNNCExternalFunction nnc_trunc(
3089 "nnc_aten_trunc",
3090 nnc_aten_trunc);
3091 const static RegisterNNCExternalFunction nnc_fix("nnc_aten_fix", nnc_aten_fix);
3092 const static RegisterNNCExternalFunction nnc_heaviside(
3093 "nnc_aten_heaviside",
3094 nnc_aten_heaviside);
3095 const static RegisterNNCExternalFunction nnc_hspmm(
3096 "nnc_aten_hspmm",
3097 nnc_aten_hspmm);
3098 const static RegisterNNCExternalFunction nnc_bitwise_and(
3099 "nnc_aten_bitwise_and",
3100 nnc_aten_bitwise_and);
3101 const static RegisterNNCExternalFunction nnc_bitwise_or(
3102 "nnc_aten_bitwise_or",
3103 nnc_aten_bitwise_or);
3104 const static RegisterNNCExternalFunction nnc_bitwise_xor(
3105 "nnc_aten_bitwise_xor",
3106 nnc_aten_bitwise_xor);
3107 const static RegisterNNCExternalFunction nnc_ne("nnc_aten_ne", nnc_aten_ne);
3108 const static RegisterNNCExternalFunction nnc_not_equal(
3109 "nnc_aten_not_equal",
3110 nnc_aten_not_equal);
3111 const static RegisterNNCExternalFunction nnc_eq("nnc_aten_eq", nnc_aten_eq);
3112 const static RegisterNNCExternalFunction nnc_ge("nnc_aten_ge", nnc_aten_ge);
3113 const static RegisterNNCExternalFunction nnc_greater_equal(
3114 "nnc_aten_greater_equal",
3115 nnc_aten_greater_equal);
3116 const static RegisterNNCExternalFunction nnc_le("nnc_aten_le", nnc_aten_le);
3117 const static RegisterNNCExternalFunction nnc_less_equal(
3118 "nnc_aten_less_equal",
3119 nnc_aten_less_equal);
3120 const static RegisterNNCExternalFunction nnc_gt("nnc_aten_gt", nnc_aten_gt);
3121 const static RegisterNNCExternalFunction nnc_greater(
3122 "nnc_aten_greater",
3123 nnc_aten_greater);
3124 const static RegisterNNCExternalFunction nnc_lt("nnc_aten_lt", nnc_aten_lt);
3125 const static RegisterNNCExternalFunction nnc_less(
3126 "nnc_aten_less",
3127 nnc_aten_less);
3128 const static RegisterNNCExternalFunction nnc_take(
3129 "nnc_aten_take",
3130 nnc_aten_take);
3131 const static RegisterNNCExternalFunction nnc_masked_select(
3132 "nnc_aten_masked_select",
3133 nnc_aten_masked_select);
3134 const static RegisterNNCExternalFunction nnc_nonzero(
3135 "nnc_aten_nonzero",
3136 nnc_aten_nonzero);
3137 const static RegisterNNCExternalFunction nnc_orgqr(
3138 "nnc_aten_orgqr",
3139 nnc_aten_orgqr);
3140 const static RegisterNNCExternalFunction nnc_lu_solve(
3141 "nnc_aten_lu_solve",
3142 nnc_aten_lu_solve);
3143 const static RegisterNNCExternalFunction nnc_lgamma(
3144 "nnc_aten_lgamma",
3145 nnc_aten_lgamma);
3146 const static RegisterNNCExternalFunction nnc_digamma(
3147 "nnc_aten_digamma",
3148 nnc_aten_digamma);
3149 const static RegisterNNCExternalFunction nnc_erfinv(
3150 "nnc_aten_erfinv",
3151 nnc_aten_erfinv);
3152 const static RegisterNNCExternalFunction nnc_i0("nnc_aten_i0", nnc_aten_i0);
3153 const static RegisterNNCExternalFunction nnc_sign(
3154 "nnc_aten_sign",
3155 nnc_aten_sign);
3156 const static RegisterNNCExternalFunction nnc_signbit(
3157 "nnc_aten_signbit",
3158 nnc_aten_signbit);
3159 const static RegisterNNCExternalFunction nnc_atan2(
3160 "nnc_aten_atan2",
3161 nnc_aten_atan2);
3162 const static RegisterNNCExternalFunction nnc_lerp(
3163 "nnc_aten_lerp",
3164 nnc_aten_lerp);
3165 const static RegisterNNCExternalFunction nnc_fmod(
3166 "nnc_aten_fmod",
3167 nnc_aten_fmod);
3168 const static RegisterNNCExternalFunction nnc_hypot(
3169 "nnc_aten_hypot",
3170 nnc_aten_hypot);
3171 const static RegisterNNCExternalFunction nnc_igamma(
3172 "nnc_aten_igamma",
3173 nnc_aten_igamma);
3174 const static RegisterNNCExternalFunction nnc_igammac(
3175 "nnc_aten_igammac",
3176 nnc_aten_igammac);
3177 const static RegisterNNCExternalFunction nnc_nextafter(
3178 "nnc_aten_nextafter",
3179 nnc_aten_nextafter);
3180 const static RegisterNNCExternalFunction nnc_remainder(
3181 "nnc_aten_remainder",
3182 nnc_aten_remainder);
3183 const static RegisterNNCExternalFunction nnc_fmin(
3184 "nnc_aten_fmin",
3185 nnc_aten_fmin);
3186 const static RegisterNNCExternalFunction nnc_fmax(
3187 "nnc_aten_fmax",
3188 nnc_aten_fmax);
3189 const static RegisterNNCExternalFunction nnc_maximum(
3190 "nnc_aten_maximum",
3191 nnc_aten_maximum);
3192 const static RegisterNNCExternalFunction nnc_max("nnc_aten_max", nnc_aten_max);
3193 const static RegisterNNCExternalFunction nnc_minimum(
3194 "nnc_aten_minimum",
3195 nnc_aten_minimum);
3196 const static RegisterNNCExternalFunction nnc_min("nnc_aten_min", nnc_aten_min);
3197 const static RegisterNNCExternalFunction nnc_msort(
3198 "nnc_aten_msort",
3199 nnc_aten_msort);
3200 const static RegisterNNCExternalFunction nnc_pow("nnc_aten_pow", nnc_aten_pow);
3201 const static RegisterNNCExternalFunction nnc_float_power(
3202 "nnc_aten_float_power",
3203 nnc_aten_float_power);
3204 const static RegisterNNCExternalFunction nnc_hardsigmoid(
3205 "nnc_aten_hardsigmoid",
3206 nnc_aten_hardsigmoid);
3207 const static RegisterNNCExternalFunction nnc_hardswish(
3208 "nnc_aten_hardswish",
3209 nnc_aten_hardswish);
3210 const static RegisterNNCExternalFunction nnc_log_sigmoid(
3211 "nnc_aten_log_sigmoid",
3212 nnc_aten_log_sigmoid);
3213 const static RegisterNNCExternalFunction nnc_log_sigmoid_backward(
3214 "nnc_aten_log_sigmoid_backward",
3215 nnc_aten_log_sigmoid_backward);
3216 const static RegisterNNCExternalFunction nnc_adaptive_avg_pool3d_backward(
3217 "nnc_aten_adaptive_avg_pool3d_backward",
3218 nnc_aten_adaptive_avg_pool3d_backward);
3219 const static RegisterNNCExternalFunction nnc_adaptive_max_pool2d_backward(
3220 "nnc_aten_adaptive_max_pool2d_backward",
3221 nnc_aten_adaptive_max_pool2d_backward);
3222 const static RegisterNNCExternalFunction nnc_adaptive_max_pool3d_backward(
3223 "nnc_aten_adaptive_max_pool3d_backward",
3224 nnc_aten_adaptive_max_pool3d_backward);
3225 const static RegisterNNCExternalFunction nnc_sigmoid_backward(
3226 "nnc_aten_sigmoid_backward",
3227 nnc_aten_sigmoid_backward);
3228 const static RegisterNNCExternalFunction nnc_tanh_backward(
3229 "nnc_aten_tanh_backward",
3230 nnc_aten_tanh_backward);
3231 const static RegisterNNCExternalFunction nnc_isposinf(
3232 "nnc_aten_isposinf",
3233 nnc_aten_isposinf);
3234 const static RegisterNNCExternalFunction nnc_isneginf(
3235 "nnc_aten_isneginf",
3236 nnc_aten_isneginf);
3237 const static RegisterNNCExternalFunction nnc_special_entr(
3238 "nnc_aten_special_entr",
3239 nnc_aten_special_entr);
3240 const static RegisterNNCExternalFunction nnc_special_expm1(
3241 "nnc_aten_special_expm1",
3242 nnc_aten_special_expm1);
3243 const static RegisterNNCExternalFunction nnc_special_exp2(
3244 "nnc_aten_special_exp2",
3245 nnc_aten_special_exp2);
3246 const static RegisterNNCExternalFunction nnc_special_gammaln(
3247 "nnc_aten_special_gammaln",
3248 nnc_aten_special_gammaln);
3249 const static RegisterNNCExternalFunction nnc_special_erf(
3250 "nnc_aten_special_erf",
3251 nnc_aten_special_erf);
3252 const static RegisterNNCExternalFunction nnc_special_erfc(
3253 "nnc_aten_special_erfc",
3254 nnc_aten_special_erfc);
3255 const static RegisterNNCExternalFunction nnc_special_erfinv(
3256 "nnc_aten_special_erfinv",
3257 nnc_aten_special_erfinv);
3258 const static RegisterNNCExternalFunction nnc_special_xlog1py(
3259 "nnc_aten_special_xlog1py",
3260 nnc_aten_special_xlog1py);
3261 const static RegisterNNCExternalFunction nnc_special_i0e(
3262 "nnc_aten_special_i0e",
3263 nnc_aten_special_i0e);
3264 const static RegisterNNCExternalFunction nnc_special_expit(
3265 "nnc_aten_special_expit",
3266 nnc_aten_special_expit);
3267 const static RegisterNNCExternalFunction nnc_linalg_cholesky(
3268 "nnc_aten_linalg_cholesky",
3269 nnc_aten_linalg_cholesky);
3270 const static RegisterNNCExternalFunction nnc_linalg_det(
3271 "nnc_aten_linalg_det",
3272 nnc_aten_linalg_det);
3273 const static RegisterNNCExternalFunction nnc_linalg_eigvals(
3274 "nnc_aten_linalg_eigvals",
3275 nnc_aten_linalg_eigvals);
3276 const static RegisterNNCExternalFunction nnc_linalg_householder_product(
3277 "nnc_aten_linalg_householder_product",
3278 nnc_aten_linalg_householder_product);
3279 const static RegisterNNCExternalFunction nnc_linalg_inv(
3280 "nnc_aten_linalg_inv",
3281 nnc_aten_linalg_inv);
3282 const static RegisterNNCExternalFunction nnc_inner(
3283 "nnc_aten_inner",
3284 nnc_aten_inner);
3285 const static RegisterNNCExternalFunction nnc_outer(
3286 "nnc_aten_outer",
3287 nnc_aten_outer);
3288 const static RegisterNNCExternalFunction nnc_ger("nnc_aten_ger", nnc_aten_ger);
3289 const static RegisterNNCExternalFunction nnc_linalg_svdvals(
3290 "nnc_aten_linalg_svdvals",
3291 nnc_aten_linalg_svdvals);
3292 const static RegisterNNCExternalFunction nnc_linalg_solve(
3293 "nnc_aten_linalg_solve",
3294 nnc_aten_linalg_solve);
3295 #endif
3296
3297 #ifdef C10_MOBILE
3298 } // extern "C"
3299 #endif
3300
3301 } // namespace torch::jit::tensorexpr
3302