Skip to content

Commit b1cf3ad

Browse files
gchananfacebook-github-bot
authored andcommitted
More Declarations.cwrap functions moved to native, mainly LAPACK, sim… (pytorch#13194)
Summary: …ple math. Pull Request resolved: pytorch#13194 Reviewed By: ezyang Differential Revision: D12811972 Pulled By: gchanan fbshipit-source-id: 461beb5efa2b6aba0808d2419eb7eb3153d18d15
1 parent dbab9b7 commit b1cf3ad

File tree

8 files changed

+925
-191
lines changed

8 files changed

+925
-191
lines changed

aten/src/ATen/Declarations.cwrap

+56-34
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,8 @@
338338
- THTensor* src
339339
]]
340340
[[
341-
name: gather
341+
name: _th_gather
342+
cname: gather
342343
variants:
343344
- method
344345
- function
@@ -1137,7 +1138,8 @@
11371138
- THTensor* self
11381139
]]
11391140
[[
1140-
name: lgamma
1141+
name: _th_lgamma
1142+
cname: lgamma
11411143
types:
11421144
- floating_point
11431145
backends:
@@ -1166,7 +1168,8 @@
11661168
- THTensor* self
11671169
]]
11681170
[[
1169-
name: digamma
1171+
name: _th_digamma
1172+
cname: digamma
11701173
types:
11711174
- floating_point
11721175
backends:
@@ -1195,7 +1198,8 @@
11951198
- THTensor* self
11961199
]]
11971200
[[
1198-
name: polygamma
1201+
name: _th_polygamma
1202+
cname: polygamma
11991203
types:
12001204
- floating_point
12011205
backends:
@@ -1423,7 +1427,8 @@
14231427
- THTensor* self
14241428
]]
14251429
[[
1426-
name: erfinv
1430+
name: _th_erfinv
1431+
cname: erfinv
14271432
types:
14281433
- floating_point
14291434
backends:
@@ -1536,7 +1541,8 @@
15361541
- THTensor* self
15371542
]]
15381543
[[
1539-
name: frac
1544+
name: _th_frac
1545+
cname: frac
15401546
types:
15411547
- floating_point
15421548
backends:
@@ -1696,7 +1702,7 @@
16961702
- real maxnorm
16971703
]]
16981704
[[
1699-
name: dist
1705+
name: _th_dist
17001706
types:
17011707
- floating_point
17021708
backends:
@@ -1716,7 +1722,7 @@
17161722
default: AS_REAL(2)
17171723
]]
17181724
[[
1719-
name: reciprocal
1725+
name: _th_reciprocal
17201726
types:
17211727
- floating_point
17221728
backends:
@@ -1748,7 +1754,7 @@
17481754
- THTensor* self
17491755
]]
17501756
[[
1751-
name: neg
1757+
name: _th_neg
17521758
backends:
17531759
- CPU
17541760
- CUDA
@@ -1776,7 +1782,7 @@
17761782
- THTensor* self
17771783
]]
17781784
[[
1779-
name: atan2
1785+
name: _th_atan2
17801786
types:
17811787
- floating_point
17821788
backends:
@@ -1869,7 +1875,7 @@
18691875
- THTensor* exponent
18701876
]]
18711877
[[
1872-
name: lerp
1878+
name: _th_lerp
18731879
types:
18741880
- floating_point
18751881
backends:
@@ -1943,7 +1949,8 @@
19431949
- long steps
19441950
]]
19451951
[[
1946-
name: histc
1952+
name: _th_histc
1953+
cname: histc
19471954
types:
19481955
- Float
19491956
- Double
@@ -1998,7 +2005,8 @@
19982005
wrap_dim: self
19992006
]]
20002007
[[
2001-
name: sign
2008+
name: _th_sign
2009+
cname: sign
20022010
variants:
20032011
- method
20042012
- function
@@ -2027,7 +2035,7 @@
20272035
- THTensor* self
20282036
]]
20292037
[[
2030-
name: fmod
2038+
name: _th_fmod
20312039
return: argument 0
20322040
variants:
20332041
- method
@@ -2064,7 +2072,7 @@
20642072
- THTensor* other
20652073
]]
20662074
[[
2067-
name: remainder
2075+
name: _th_remainder
20682076
return: argument 0
20692077
variants:
20702078
- method
@@ -2465,7 +2473,8 @@
24652473
- THTensor* batch2
24662474
]]
24672475
[[
2468-
name: addcmul
2476+
name: _th_addcmul
2477+
cname: addcmul
24692478
variants:
24702479
- method
24712480
- function
@@ -2507,7 +2516,8 @@
25072516
- THSTensor* tensor2
25082517
]]
25092518
[[
2510-
name: addcdiv
2519+
name: _th_addcdiv
2520+
cname: addcdiv
25112521
variants:
25122522
- method
25132523
- function
@@ -2557,7 +2567,8 @@
25572567
- THTensor* A
25582568
]]
25592569
[[
2560-
name: gels
2570+
name: _th_gels
2571+
cname: gels
25612572
types:
25622573
- Float
25632574
- Double
@@ -2577,7 +2588,8 @@
25772588
- THTensor* A
25782589
]]
25792590
[[
2580-
name: trtrs
2591+
name: _th_trtrs
2592+
cname: trtrs
25812593
types:
25822594
- Float
25832595
- Double
@@ -2609,7 +2621,7 @@
26092621
default: N
26102622
]]
26112623
[[
2612-
name: symeig
2624+
name: _th_symeig
26132625
cname: syev
26142626
types:
26152627
- Float
@@ -2637,7 +2649,7 @@
26372649
default: U
26382650
]]
26392651
[[
2640-
name: eig
2652+
name: _th_eig
26412653
cname: geev
26422654
types:
26432655
- Float
@@ -2661,7 +2673,7 @@
26612673
default: N
26622674
]]
26632675
[[
2664-
name: svd
2676+
name: _th_svd
26652677
cname: gesdd
26662678
types:
26672679
- Float
@@ -2707,7 +2719,8 @@
27072719
- THTensor* self
27082720
]]
27092721
[[
2710-
name: potrf
2722+
name: _th_potrf
2723+
cname: potrf
27112724
types:
27122725
- Float
27132726
- Double
@@ -2728,7 +2741,8 @@
27282741
default: U
27292742
]]
27302743
[[
2731-
name: potrs
2744+
name: _th_potrs
2745+
cname: potrs
27322746
types:
27332747
- Float
27342748
- Double
@@ -2750,7 +2764,8 @@
27502764
default: U
27512765
]]
27522766
[[
2753-
name: potri
2767+
name: _th_potri
2768+
cname: potri
27542769
types:
27552770
- Float
27562771
- Double
@@ -2771,7 +2786,8 @@
27712786
default: U
27722787
]]
27732788
[[
2774-
name: pstrf
2789+
name: _th_pstrf
2790+
cname: pstrf
27752791
types:
27762792
- Float
27772793
- Double
@@ -2798,7 +2814,8 @@
27982814
res2 -= 1; // LAPACK returns 1-indexed pivots
27992815
]]
28002816
[[
2801-
name: qr
2817+
name: _th_qr
2818+
cname: qr
28022819
types:
28032820
- Float
28042821
- Double
@@ -2817,7 +2834,8 @@
28172834
- THTensor* self
28182835
]]
28192836
[[
2820-
name: geqrf
2837+
name: _th_geqrf
2838+
cname: geqrf
28212839
types:
28222840
- Float
28232841
- Double
@@ -2836,7 +2854,8 @@
28362854
- THTensor* self
28372855
]]
28382856
[[
2839-
name: orgqr
2857+
name: _th_orgqr
2858+
cname: orgqr
28402859
types:
28412860
- Float
28422861
- Double
@@ -2853,7 +2872,8 @@
28532872
- THTensor* input2
28542873
]]
28552874
[[
2856-
name: ormqr
2875+
name: _th_ormqr
2876+
cname: ormqr
28572877
types:
28582878
- Float
28592879
- Double
@@ -2879,7 +2899,7 @@
28792899
default: N
28802900
]]
28812901
[[
2882-
name: btrifact
2902+
name: _th_btrifact
28832903
cname: btrifact
28842904
types:
28852905
- floating_point
@@ -2902,7 +2922,7 @@
29022922
- THTensor* self
29032923
]]
29042924
[[
2905-
name: btrifact_with_info
2925+
name: _th_btrifact_with_info
29062926
cname: btrifact
29072927
types:
29082928
- floating_point
@@ -2926,7 +2946,7 @@
29262946
- THTensor* self
29272947
]]
29282948
[[
2929-
name: btrisolve
2949+
name: _th_btrisolve
29302950
cname: btrisolve
29312951
types:
29322952
- floating_point
@@ -2974,7 +2994,8 @@
29742994
- int64_t to
29752995
]]
29762996
[[
2977-
name: multinomial
2997+
name: _th_multinomial
2998+
cname: multinomial
29782999
types:
29793000
- floating_point
29803001
backends:
@@ -3016,6 +3037,7 @@
30163037
]]
30173038
[[
30183039
name: normal
3040+
cname: normal
30193041
types:
30203042
- floating_point
30213043
backends:

0 commit comments

Comments
 (0)