xiaoanyu123 commited on
Commit
6698547
·
verified ·
1 Parent(s): 3dde010

Add files using upload-large-folder tool

Browse files
Files changed (20) hide show
  1. pythonProject/.venv/Lib/site-packages/numpy/distutils/checks/cpu_vxe2.c +21 -0
  2. pythonProject/.venv/Lib/site-packages/numpy/distutils/checks/cpu_xop.c +12 -0
  3. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_both.f90 +57 -0
  4. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_compound.f90 +15 -0
  5. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_integer.f90 +22 -0
  6. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_non_compound.f90 +23 -0
  7. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_real.f90 +23 -0
  8. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/quoted_character/foo.f +14 -0
  9. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/AB.inc +1 -0
  10. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/assignOnlyModule.f90 +25 -0
  11. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/datonly.f90 +17 -0
  12. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f77comments.f +26 -0
  13. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f77fixedform.f95 +5 -0
  14. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f90continuation.f90 +9 -0
  15. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/incfile.f90 +5 -0
  16. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/inout.f90 +9 -0
  17. pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 +5 -0
  18. pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_lu_solve_ops.h +39 -0
  19. pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_matmul.h +39 -0
  20. pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_matmul_compositeimplicitautograd_dispatch.h +25 -0
pythonProject/.venv/Lib/site-packages/numpy/distutils/checks/cpu_vxe2.c ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #if (__VEC__ < 10303) || (__ARCH__ < 13)
2
+ #error VXE2 not supported
3
+ #endif
4
+
5
+ #include <vecintrin.h>
6
+
7
+ int main(int argc, char **argv)
8
+ {
9
+ int val;
10
+ __vector signed short large = { 'a', 'b', 'c', 'a', 'g', 'h', 'g', 'o' };
11
+ __vector signed short search = { 'g', 'h', 'g', 'o' };
12
+ __vector unsigned char len = { 0 };
13
+ __vector unsigned char res = vec_search_string_cc(large, search, len, &val);
14
+ __vector float x = vec_xl(argc, (float*)argv);
15
+ __vector int i = vec_signed(x);
16
+
17
+ i = vec_srdb(vec_sldb(i, i, 2), i, 3);
18
+ val += (int)vec_extract(res, 1);
19
+ val += vec_extract(i, 0);
20
+ return val;
21
+ }
pythonProject/.venv/Lib/site-packages/numpy/distutils/checks/cpu_xop.c ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #include <immintrin.h>
2
+ #ifdef _MSC_VER
3
+ #include <ammintrin.h>
4
+ #else
5
+ #include <x86intrin.h>
6
+ #endif
7
+
8
+ int main(void)
9
+ {
10
+ __m128i a = _mm_comge_epu32(_mm_setzero_si128(), _mm_setzero_si128());
11
+ return _mm_cvtsi128_si32(a);
12
+ }
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_both.f90 ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that parameters are correct intercepted.
2
+ ! Constants with comma separations are commonly
3
+ ! used, for instance Pi = 3._dp
4
+ subroutine foo(x)
5
+ implicit none
6
+ integer, parameter :: sp = selected_real_kind(6)
7
+ integer, parameter :: dp = selected_real_kind(15)
8
+ integer, parameter :: ii = selected_int_kind(9)
9
+ integer, parameter :: il = selected_int_kind(18)
10
+ real(dp), intent(inout) :: x
11
+ dimension x(3)
12
+ real(sp), parameter :: three_s = 3._sp
13
+ real(dp), parameter :: three_d = 3._dp
14
+ integer(ii), parameter :: three_i = 3_ii
15
+ integer(il), parameter :: three_l = 3_il
16
+ x(1) = x(1) + x(2) * three_s * three_i + x(3) * three_d * three_l
17
+ x(2) = x(2) * three_s
18
+ x(3) = x(3) * three_l
19
+ return
20
+ end subroutine
21
+
22
+
23
+ subroutine foo_no(x)
24
+ implicit none
25
+ integer, parameter :: sp = selected_real_kind(6)
26
+ integer, parameter :: dp = selected_real_kind(15)
27
+ integer, parameter :: ii = selected_int_kind(9)
28
+ integer, parameter :: il = selected_int_kind(18)
29
+ real(dp), intent(inout) :: x
30
+ dimension x(3)
31
+ real(sp), parameter :: three_s = 3.
32
+ real(dp), parameter :: three_d = 3.
33
+ integer(ii), parameter :: three_i = 3
34
+ integer(il), parameter :: three_l = 3
35
+ x(1) = x(1) + x(2) * three_s * three_i + x(3) * three_d * three_l
36
+ x(2) = x(2) * three_s
37
+ x(3) = x(3) * three_l
38
+ return
39
+ end subroutine
40
+
41
+ subroutine foo_sum(x)
42
+ implicit none
43
+ integer, parameter :: sp = selected_real_kind(6)
44
+ integer, parameter :: dp = selected_real_kind(15)
45
+ integer, parameter :: ii = selected_int_kind(9)
46
+ integer, parameter :: il = selected_int_kind(18)
47
+ real(dp), intent(inout) :: x
48
+ dimension x(3)
49
+ real(sp), parameter :: three_s = 2._sp + 1._sp
50
+ real(dp), parameter :: three_d = 1._dp + 2._dp
51
+ integer(ii), parameter :: three_i = 2_ii + 1_ii
52
+ integer(il), parameter :: three_l = 1_il + 2_il
53
+ x(1) = x(1) + x(2) * three_s * three_i + x(3) * three_d * three_l
54
+ x(2) = x(2) * three_s
55
+ x(3) = x(3) * three_l
56
+ return
57
+ end subroutine
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_compound.f90 ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that parameters are correct intercepted.
2
+ ! Constants with comma separations are commonly
3
+ ! used, for instance Pi = 3._dp
4
+ subroutine foo_compound_int(x)
5
+ implicit none
6
+ integer, parameter :: ii = selected_int_kind(9)
7
+ integer(ii), intent(inout) :: x
8
+ dimension x(3)
9
+ integer(ii), parameter :: three = 3_ii
10
+ integer(ii), parameter :: two = 2_ii
11
+ integer(ii), parameter :: six = three * 1_ii * two
12
+
13
+ x(1) = x(1) + x(2) + x(3) * six
14
+ return
15
+ end subroutine
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_integer.f90 ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that parameters are correct intercepted.
2
+ ! Constants with comma separations are commonly
3
+ ! used, for instance Pi = 3._dp
4
+ subroutine foo_int(x)
5
+ implicit none
6
+ integer, parameter :: ii = selected_int_kind(9)
7
+ integer(ii), intent(inout) :: x
8
+ dimension x(3)
9
+ integer(ii), parameter :: three = 3_ii
10
+ x(1) = x(1) + x(2) + x(3) * three
11
+ return
12
+ end subroutine
13
+
14
+ subroutine foo_long(x)
15
+ implicit none
16
+ integer, parameter :: ii = selected_int_kind(18)
17
+ integer(ii), intent(inout) :: x
18
+ dimension x(3)
19
+ integer(ii), parameter :: three = 3_ii
20
+ x(1) = x(1) + x(2) + x(3) * three
21
+ return
22
+ end subroutine
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_non_compound.f90 ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that parameters are correct intercepted.
2
+ ! Specifically that types of constants without
3
+ ! compound kind specs are correctly inferred
4
+ ! adapted Gibbs iteration code from pymc
5
+ ! for this test case
6
+ subroutine foo_non_compound_int(x)
7
+ implicit none
8
+ integer, parameter :: ii = selected_int_kind(9)
9
+
10
+ integer(ii) maxiterates
11
+ parameter (maxiterates=2)
12
+
13
+ integer(ii) maxseries
14
+ parameter (maxseries=2)
15
+
16
+ integer(ii) wasize
17
+ parameter (wasize=maxiterates*maxseries)
18
+ integer(ii), intent(inout) :: x
19
+ dimension x(wasize)
20
+
21
+ x(1) = x(1) + x(2) + x(3) + x(4) * wasize
22
+ return
23
+ end subroutine
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/parameter/constant_real.f90 ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that parameters are correct intercepted.
2
+ ! Constants with comma separations are commonly
3
+ ! used, for instance Pi = 3._dp
4
+ subroutine foo_single(x)
5
+ implicit none
6
+ integer, parameter :: rp = selected_real_kind(6)
7
+ real(rp), intent(inout) :: x
8
+ dimension x(3)
9
+ real(rp), parameter :: three = 3._rp
10
+ x(1) = x(1) + x(2) + x(3) * three
11
+ return
12
+ end subroutine
13
+
14
+ subroutine foo_double(x)
15
+ implicit none
16
+ integer, parameter :: rp = selected_real_kind(15)
17
+ real(rp), intent(inout) :: x
18
+ dimension x(3)
19
+ real(rp), parameter :: three = 3._rp
20
+ x(1) = x(1) + x(2) + x(3) * three
21
+ return
22
+ end subroutine
23
+
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/quoted_character/foo.f ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ SUBROUTINE FOO(OUT1, OUT2, OUT3, OUT4, OUT5, OUT6)
2
+ CHARACTER SINGLE, DOUBLE, SEMICOL, EXCLA, OPENPAR, CLOSEPAR
3
+ PARAMETER (SINGLE="'", DOUBLE='"', SEMICOL=';', EXCLA="!",
4
+ 1 OPENPAR="(", CLOSEPAR=")")
5
+ CHARACTER OUT1, OUT2, OUT3, OUT4, OUT5, OUT6
6
+ Cf2py intent(out) OUT1, OUT2, OUT3, OUT4, OUT5, OUT6
7
+ OUT1 = SINGLE
8
+ OUT2 = DOUBLE
9
+ OUT3 = SEMICOL
10
+ OUT4 = EXCLA
11
+ OUT5 = OPENPAR
12
+ OUT6 = CLOSEPAR
13
+ RETURN
14
+ END
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/AB.inc ADDED
@@ -0,0 +1 @@
 
 
1
+ real(8) b, n, m
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/assignOnlyModule.f90 ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MODULE MOD_TYPES
2
+ INTEGER, PARAMETER :: SP = SELECTED_REAL_KIND(6, 37)
3
+ INTEGER, PARAMETER :: DP = SELECTED_REAL_KIND(15, 307)
4
+ END MODULE
5
+ !
6
+ MODULE F_GLOBALS
7
+ USE MOD_TYPES
8
+ IMPLICIT NONE
9
+ INTEGER, PARAMETER :: N_MAX = 16
10
+ INTEGER, PARAMETER :: I_MAX = 18
11
+ INTEGER, PARAMETER :: J_MAX = 72
12
+ REAL(SP) :: XREF
13
+ END MODULE F_GLOBALS
14
+ !
15
+ SUBROUTINE DUMMY ()
16
+ !
17
+ USE F_GLOBALS
18
+ USE MOD_TYPES
19
+ IMPLICIT NONE
20
+ !
21
+ REAL(SP) :: MINIMAL
22
+ MINIMAL = 0.01*XREF
23
+ RETURN
24
+ !
25
+ END SUBROUTINE DUMMY
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/datonly.f90 ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ module datonly
2
+ implicit none
3
+ integer, parameter :: max_value = 100
4
+ real, dimension(:), allocatable :: data_array
5
+ end module datonly
6
+
7
+ module dat
8
+ implicit none
9
+ integer, parameter :: max_= 1009
10
+ end module dat
11
+
12
+ subroutine simple_subroutine(ain, aout)
13
+ use dat, only: max_
14
+ integer, intent(in) :: ain
15
+ integer, intent(out) :: aout
16
+ aout = ain + max_
17
+ end subroutine simple_subroutine
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f77comments.f ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ SUBROUTINE TESTSUB(
2
+ & INPUT1, INPUT2, !Input
3
+ & OUTPUT1, OUTPUT2) !Output
4
+
5
+ IMPLICIT NONE
6
+ INTEGER, INTENT(IN) :: INPUT1, INPUT2
7
+ INTEGER, INTENT(OUT) :: OUTPUT1, OUTPUT2
8
+
9
+ OUTPUT1 = INPUT1 + INPUT2
10
+ OUTPUT2 = INPUT1 * INPUT2
11
+
12
+ RETURN
13
+ END SUBROUTINE TESTSUB
14
+
15
+ SUBROUTINE TESTSUB2(OUTPUT)
16
+ IMPLICIT NONE
17
+ INTEGER, PARAMETER :: N = 10 ! Array dimension
18
+ REAL, INTENT(OUT) :: OUTPUT(N)
19
+ INTEGER :: I
20
+
21
+ DO I = 1, N
22
+ OUTPUT(I) = I * 2.0
23
+ END DO
24
+
25
+ RETURN
26
+ END
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f77fixedform.f95 ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ C This is an invalid file, but it does compile with -ffixed-form
2
+ subroutine mwe(
3
+ & x)
4
+ real x
5
+ end subroutine mwe
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/f90continuation.f90 ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ SUBROUTINE TESTSUB(INPUT1, & ! Hello
2
+ ! commenty
3
+ INPUT2, OUTPUT1, OUTPUT2) ! more comments
4
+ INTEGER, INTENT(IN) :: INPUT1, INPUT2
5
+ INTEGER, INTENT(OUT) :: OUTPUT1, OUTPUT2
6
+ OUTPUT1 = INPUT1 + &
7
+ INPUT2
8
+ OUTPUT2 = INPUT1 * INPUT2
9
+ END SUBROUTINE TESTSUB
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/incfile.f90 ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ function add(n,m) result(b)
2
+ implicit none
3
+ include 'AB.inc'
4
+ b = n + m
5
+ end function add
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/inout.f90 ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ! Check that intent(in out) translates as intent(inout).
2
+ ! The separation seems to be a common usage.
3
+ subroutine foo(x)
4
+ implicit none
5
+ real(4), intent(in out) :: x
6
+ dimension x(3)
7
+ x(1) = x(1) + x(2) + x(3)
8
+ return
9
+ end
pythonProject/.venv/Lib/site-packages/numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ subroutine inquire_next(IU)
2
+ IMPLICIT NONE
3
+ integer :: IU
4
+ !f2py intent(in) IU
5
+ end subroutine
pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_lu_solve_ops.h ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ // @generated by torchgen/gen.py from Operator.h
4
+
5
+ #include <tuple>
6
+ #include <vector>
7
+
8
+ // Forward declarations of any types needed in the operator signatures.
9
+ // We can't directly include these classes because it will cause circular include dependencies.
10
+ // This file is included by TensorBody.h, which defines the Tensor class.
11
+ #include <ATen/core/ATen_fwd.h>
12
+
13
+ namespace at {
14
+ namespace _ops {
15
+
16
+
17
+ struct TORCH_API linalg_lu_solve {
18
+ using schema = at::Tensor (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool);
19
+ using ptr_schema = schema*;
20
+ // See Note [static constexpr char* members for windows NVCC]
21
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::linalg_lu_solve")
22
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "")
23
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "linalg_lu_solve(Tensor LU, Tensor pivots, Tensor B, *, bool left=True, bool adjoint=False) -> Tensor")
24
+ static at::Tensor call(const at::Tensor & LU, const at::Tensor & pivots, const at::Tensor & B, bool left, bool adjoint);
25
+ static at::Tensor redispatch(c10::DispatchKeySet dispatchKeySet, const at::Tensor & LU, const at::Tensor & pivots, const at::Tensor & B, bool left, bool adjoint);
26
+ };
27
+
28
+ struct TORCH_API linalg_lu_solve_out {
29
+ using schema = at::Tensor & (const at::Tensor &, const at::Tensor &, const at::Tensor &, bool, bool, at::Tensor &);
30
+ using ptr_schema = schema*;
31
+ // See Note [static constexpr char* members for windows NVCC]
32
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::linalg_lu_solve")
33
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "out")
34
+ STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "linalg_lu_solve.out(Tensor LU, Tensor pivots, Tensor B, *, bool left=True, bool adjoint=False, Tensor(a!) out) -> Tensor(a!)")
35
+ static at::Tensor & call(const at::Tensor & LU, const at::Tensor & pivots, const at::Tensor & B, bool left, bool adjoint, at::Tensor & out);
36
+ static at::Tensor & redispatch(c10::DispatchKeySet dispatchKeySet, const at::Tensor & LU, const at::Tensor & pivots, const at::Tensor & B, bool left, bool adjoint, at::Tensor & out);
37
+ };
38
+
39
+ }} // namespace at::_ops
pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_matmul.h ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ // @generated by torchgen/gen.py from Function.h
4
+
5
+ #include <ATen/Context.h>
6
+ #include <ATen/DeviceGuard.h>
7
+ #include <ATen/TensorUtils.h>
8
+ #include <ATen/TracerMode.h>
9
+ #include <ATen/core/Generator.h>
10
+ #include <ATen/core/Reduction.h>
11
+ #include <ATen/core/Tensor.h>
12
+ #include <c10/core/Scalar.h>
13
+ #include <c10/core/Storage.h>
14
+ #include <c10/core/TensorOptions.h>
15
+ #include <c10/util/Deprecated.h>
16
+ #include <c10/util/Optional.h>
17
+
18
+
19
+
20
+ #include <ATen/ops/linalg_matmul_ops.h>
21
+
22
+ namespace at {
23
+
24
+
25
+ // aten::linalg_matmul(Tensor self, Tensor other) -> Tensor
26
+ inline at::Tensor linalg_matmul(const at::Tensor & self, const at::Tensor & other) {
27
+ return at::_ops::linalg_matmul::call(self, other);
28
+ }
29
+
30
+ // aten::linalg_matmul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!)
31
+ inline at::Tensor & linalg_matmul_out(at::Tensor & out, const at::Tensor & self, const at::Tensor & other) {
32
+ return at::_ops::linalg_matmul_out::call(self, other, out);
33
+ }
34
+ // aten::linalg_matmul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!)
35
+ inline at::Tensor & linalg_matmul_outf(const at::Tensor & self, const at::Tensor & other, at::Tensor & out) {
36
+ return at::_ops::linalg_matmul_out::call(self, other, out);
37
+ }
38
+
39
+ }
pythonProject/.venv/Lib/site-packages/torch/include/ATen/ops/linalg_matmul_compositeimplicitautograd_dispatch.h ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+ // @generated by torchgen/gen.py from DispatchKeyFunction.h
3
+
4
+ // NB: The implementing C++ file is RegisterDispatchKey.cpp
5
+
6
+ // The only #includes we need are for custom classes that have defaults in the C++ API
7
+ #include <c10/core/MemoryFormat.h>
8
+ #include <c10/core/Scalar.h>
9
+ #include <ATen/core/Reduction.h>
10
+
11
+ // Forward declarations of any types needed in the operator signatures.
12
+ // We can't directly include these classes because it will cause circular include dependencies.
13
+ // This file is included by TensorBody.h, which defines the Tensor class.
14
+ #include <ATen/core/ATen_fwd.h>
15
+
16
+ namespace at {
17
+
18
+ namespace compositeimplicitautograd {
19
+
20
+ TORCH_API at::Tensor linalg_matmul(const at::Tensor & self, const at::Tensor & other);
21
+ TORCH_API at::Tensor & linalg_matmul_out(at::Tensor & out, const at::Tensor & self, const at::Tensor & other);
22
+ TORCH_API at::Tensor & linalg_matmul_outf(const at::Tensor & self, const at::Tensor & other, at::Tensor & out);
23
+
24
+ } // namespace compositeimplicitautograd
25
+ } // namespace at