8000 Add torch linalg data · data-apis/array-api-comparison@bb9922d · GitHub
[go: up one dir, main page]
Skip to content

Commit bb9922d

Browse files
kgrytekgryte
committed
Add torch linalg data
1 parent 1364673 commit bb9922d

File tree

4 files changed

+58
-40
lines changed

4 files changed

+58
-40
lines changed

‎data/array_api.csv

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -54,25 +54,25 @@ name,numpy<sup>*</sup>,cupy,dask,jax,mxnet,pytorch,sparse,tensorflow
5454
"isnan","&#128640","&#128640","&#128640","&#128640","","&#128640","",""
5555
"less","&#128640","&#128640","&#128640","&#128640","","&#128640","",""
5656
"less_equal","&#128640","&#128640","","&#128640","","&#128640","",""
57-
"linalg.cholesky","","&#128640","&#128640","","","","","&#128640"
57+
"linalg.cholesky","","&#128640","&#128640","","","&#128640","","&#128640"
5858
"linalg.cross","","","","","","","","&#128640"
59-
"linalg.det","","&#128640","","","","","","&#128640"
59+
"linalg.det","","&#128640","","","","&#128640","","&#128640"
6060
"linalg.diagonal","","","","","","","",""
61-
"linalg.eigh","","&#128640","","","","","","&#128640"
62-
"linalg.eigvalsh","","&#128640","","","","","","&#128640"
61+
"linalg.eigh","","&#128640","","","","&#128640","","&#128640"
62+
"linalg.eigvalsh","","&#128640","","","","&#128640","","&#128640"
6363
"linalg.einsum"," 8000 ","","","","","","",""
64-
"linalg.inv","","&#128640","&#128640","","","","","&#128640"
64+
"linalg.inv","","&#128640","&#128640","","","&#128640","","&#128640"
6565
"linalg.lstsq","","&#128640","&#128640","","","","","&#128640"
6666
"linalg.matmul","","","","","","","","&#128640"
6767
"linalg.matrix_power","","&#128640","","","","","",""
68-
"linalg.matrix_rank","","&#128640","","","","","","&#128640"
69-
"linalg.norm","","&#128640","&#128640","","","","",""
68+
"linalg.matrix_rank","","&#128640","","","","&#128640","","&#128640"
69+
"linalg.norm","","&#128640","&#128640","","","&#128640","",""
7070
"linalg.outer","","","","","","","",""
71-
"linalg.pinv","","&#128640","","","","","","&#128640"
72-
"linalg.qr","","&#128640","&#128640","","","","","&#128640"
73-
"linalg.slogdet","","&#128640","","","","","","&#128640"
74-
"linalg.solve","","&#128640","&#128640","","","","","&#128640"
75-
"linalg.svd","","&#128640","&#128640","","","","","&#128640"
71+
"linalg.pinv","","&#128640","","","","&#128640","","&#128640"
72+
"linalg.qr","","&#128640","&#128640","","","&#128640","","&#128640"
73+
"linalg.slogdet","","&#128640","","","","&#128640","","&#128640"
74+
"linalg.solve","","&#128640","&#128640","","","&#128640","","&#128640"
75+
"linalg.svd","","&#128640","&#128640","","","&#128640","","&#128640"
7676
"linalg.svdvals","","","","","","","",""
7777
"linalg.tensordot","","","","","","","",""
7878
"linalg.trace","","","","","","","","&#128640"
@@ -131,4 +131,4 @@ name,numpy<sup>*</sup>,cupy,dask,jax,mxnet,pytorch,sparse,tensorflow
131131
"where","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640"
132132
"zeros","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640"
133133
"zeros_like","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640","&#128640"
134-
"Total APIs: 132","104","96","87","90","61","92","13","36"
134+
"Total APIs: 132","104","96","87","90","61","104","13","36"

‎data/array_api.json

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@
611611
"dask": "&#128640",
612612
"jax": "",
613613
"mxnet": "",
614-
"pytorch": "",
614+
"pytorch": "&#128640",
615615
"sparse": "",
616616
"tensorflow": "&#128640"
617617
},
@@ -633,7 +633,7 @@
633633
"dask": "",
634634
"jax": "",
635635
"mxnet": "",
636-
"pytorch": "",
636+
"pytorch": "&#128640",
637637
"sparse": "",
638638
"tensorflow": "&#128640"
639639
},
@@ -655,7 +655,7 @@
655655
"dask": "",
656656
"jax": "",
657657
"mxnet": "",
658-
"pytorch": "",
658+
"pytorch": "&#128640",
659659
"sparse": "",
660660
"tensorflow": "&#128640"
661661
},
@@ -666,7 +666,7 @@
666666
"dask": "",
667667
"jax": "",
668668
"mxnet": "",
669-
"pytorch": "",
669+
"pytorch": "&#128640",
670670
"sparse": "",
671671
"tensorflow": "&#128640"
672672
},
@@ -688,7 +688,7 @@
688688
"dask": "&#128640",
689689
"jax": "",
690690
"mxnet": "",
691-
"pytorch": "",
691+
"pytorch": "&#128640",
692692
"sparse": "",
693693
"tensorflow": "&#128640"
694694
},
@@ -732,7 +732,7 @@
732732
"dask": "",
733733
"jax": "",
734734
"mxnet": "",
735-
"pytorch": "",
735+
"pytorch": "&#128640",
736736
"sparse": "",
737737
"tensorflow": "&#128640"
738738
},
@@ -743,7 +743,7 @@
743743
"dask": "&#128640",
744744
"jax": "",
745745
"mxnet": "",
746-
"pytorch": "",
746+
"pytorch": "&#128640",
747747
"sparse": "",
748748
"tensorflow": ""
749749
},
@@ -765,7 +765,7 @@
765765
"dask": "",
766766
"jax": "",
767767
"mxnet": "",
768-
"pytorch": "",
768+
"pytorch": "&#128640",
769769
"sparse": "",
770770
"tensorflow": "&#128640"
771771
},
@@ -776,7 +776,7 @@
776776
"dask": "&#128640",
777777
"jax": "",
778778
"mxnet": "",
779-
"pytorch": "",
779+
"pytorch": "&#128 10000 640",
780780
"sparse": "",
781781
"tensorflow": "&#128640"
782782
},
@@ -787,7 +787,7 @@
787787
"dask": "",
788788
"jax": "",
789789
"mxnet": "",
790-
"pytorch": "",
790+
"pytorch": "&#128640",
791791
"sparse": "",
792792
"tensorflow": "&#128640"
793793
},
@@ -798,7 +798,7 @@
798798
"dask": "&#128640",
799799
"jax": "",
800800
"mxnet": "",
801-
"pytorch": "",
801+
"pytorch": "&#128640",
802802
"sparse": "",
803803
"tensorflow": "&#128640"
804804
},
@@ -809,7 +809,7 @@
809809
"dask": "&#128640",
810810
"jax": "",
811811
"mxnet": "",
812-
"pytorch": "",
812+
"pytorch": "&#128640",
813813
"sparse": "",
814814
"tensorflow": "&#128640"
815815
},
@@ -1458,7 +1458,7 @@
14581458
"dask": 87,
14591459
"jax": 90,
14601460
"mxnet": 61,
1461-
"pytorch": 92,
1461+
"pytorch": 104,
14621462
"sparse": 13,
14631463
"tensorflow": 36
14641464
}

‎docs/array_api.html

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -644,7 +644,7 @@ <h1>Array API Function Name Compliance</h1>
644644
<td>&#128640</td>
645645
<td></td>
646646
<td></td>
647-
<td></td>
647+
<td>&#128640</td>
648648
<td></td>
649649
<td>&#128640</td>
650650
</tr>
@@ -666,7 +666,7 @@ <h1>Array API Function Name Compliance</h1>
666666
<td></td>
667667
<td></td>
668668
<td></td>
669-
<td></td>
669+
<td>&#128640</td>
670670
<td></td>
671671
<td>&#128640</td>
672672
</tr>
@@ -688,7 +688,7 @@ <h1>Array API Function Name Compliance</h1>
688688
<td></td>
689689
<td></td>
690690
<td></td>
691-
<td></td>
691+
<td>&#128640</td>
692692
<td></td>
693693
<td>&#128640</td>
694694
</tr>
@@ -699,7 +699,7 @@ <h1>Array API Function Name Compliance</h1>
699699
<td></td>
700700
<td></td>
701701
<td></td>
702-
<td></td>
702+
<td>&#128640</td>
703703
<td></td>
704704
<td>&#128640</td>
705705
</tr>
@@ -721,7 +721,7 @@ <h1>Array API Function Name Compliance</h1>
721721
<td>&#128640</td>
722722
<td></td>
723723
<td></td>
724-
<td></td>
724+
<td>&#128640</td>
725725
<td></td>
726726
<td>&#128640</td>
727727
</tr>
@@ -765,7 +765,7 @@ <h1>Array API Function Name Compliance</h1>
765765
<td></td>
766766
<td></td>
767767
<td></td>
768-
<td></td>
768+
<td>&#128640</td>
769769
<td></td>
770770
<td>&#128640</td>
771771
</tr>
@@ -776,7 +776,7 @@ <h1>Array API Function Name Compliance</h1>
776776
<td>&#128640</td>
777777
<td></td>
778778
<td></td>
779-
<td></td>
779+
<td>&#128640</td>
780780
<td></td>
781781
<td></td>
782782
</tr>
@@ -798,7 +798,7 @@ <h1>Array API Function Name Compliance</h1>
798798
<td></td>
799799
<td></td>
800800
<td></td>
801-
<td></td>
801+
<td>&#128640</td>
802802
<td></td>
803803
<td>&#128640</td>
804804
</tr>
@@ -809,7 +809,7 @@ <h1>Array API Function Name Compliance</h1>
809809
<td>&#128640</td>
810810
<td></td>
811811
<td></td>
812-
<td></td>
812+
<td>&#128640</td>
813813
<td></td>
814814
<td>&#128640</td>
815815
</tr>
@@ -820,7 +820,7 @@ <h1>Array API Function Name Compliance</h1>
820820
<td></td>
821821
<td></td>
822822
<td></td>
823-
<td></td>
823+
<td>&#128640</td>
824824
<td></td>
825825
<td>&#128640</td>
826826
</tr>
@@ -831,7 +831,7 @@ <h1>Array API Function Name Compliance</h1>
831831
<td>&#128640</td>
832832
<td></td>
833833
<td></td>
834-
<td></td>
834+
<td>&#128640</td>
835835
<td></td>
836836
<td>&#128640</td>
837837
</tr>
@@ -842,7 +842,7 @@ <h1>Array API Function Name Compliance</h1>
842842
<td>&#128640</td>
843843
<td></td>
844844
<td></td>
845-
<td></td>
845+
<td>&#128640</td>
846846
<td></td>
847847
<td>&#128640</td>
848848
</tr>
@@ -1491,7 +1491,7 @@ <h1>Array API Function Name Compliance</h1>
14911491
<td>87</td>
14921492
<td>90</td>
14931493
<td>61</td>
1494-
<td>92</td>
1494+
<td>104</td>
14951495
<td>13</td>
14961496
<td>36</td>
14971497
</tr>

‎scripts/array_api.js

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,29 @@
3030

3131
var join = require( 'path' ).join;
3232
var readJSON = require( '@stdlib/fs/read-json' ).sync;
33+
var contains = require( '@stdlib/assert/contains' );
3334
var ARRAY_API = require( './../data/raw/array_api.json' );
3435

3536

3637
// VARIABLES //
3738

39+
var TORCH_LINALG = [
40+
'torch.linalg.cholesky',
41+
'torch.linalg.cond',
42+
'torch.linalg.det',
43+
'torch.linalg.slogdet',
44+
'torch.linalg.eigh',
45+
'torch.linalg.eigvalsh',
46+
'torch.linalg.matrix_rank',
47+
'torch.linalg.norm',
48+
'torch.linalg.pinv',
49+
'torch.linalg.svd',
50+
'torch.linalg.solve',
51+
'torch.linalg.tensorinv',
52+
'torch.linalg.tensorsolve',
53+
'torch.linalg.inv',
54+
'torch.linalg.qr'
55+
];
3856
var LIBRARIES = [
3957
'numpy',
4058
'cupy',
@@ -115,7 +133,7 @@ function main() {
115133
alias = PREFIXES[ key ] + name;
116134
for ( k = 0; k < data.length; k++ ) {
117135
o = data[ k ];
118-
if ( o.name === alias ) {
136+
if ( o.name === alias || ( key === 'pytorch' && contains( TORCH_LINALG, alias ) ) ) {
119137
// TODO: remove once https://github.com/numpy/numpy/pull/18585 is merged; here, we mark with an asterisk to indicate that this is in-progress...
120138
if ( key === 'numpy' ) {
121139
key += '<sup>*</sup>';

0 commit comments

Comments
 (0)
0