Prompt48 commited on
Commit
4c8487c
·
verified ·
1 Parent(s): 34da12e

Upload edit\Qwen3-TTS-test\.venv\Lib\site-packages\torch\include\ATen\native\mkldnn\xpu\detail\oneDNN.h with huggingface_hub

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//mkldnn//xpu//detail//oneDNN.h ADDED
@@ -0,0 +1,136 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ #include <ATen/ATen.h>
4
+ #include <ATen/native/mkldnn/xpu/detail/Attr.h>
5
+ #include <ATen/native/mkldnn/xpu/detail/Utils.h>
6
+ #include <ATen/native/mkldnn/xpu/detail/oneDNNContext.h>
7
+
8
+ namespace at::native::onednn {
9
+
10
+ TORCH_API sycl::event matmul(
11
+ at::Tensor& result,
12
+ const at::Tensor& mat1,
13
+ const at::Tensor& mat2,
14
+ const at::Tensor& b_raw,
15
+ bool m2_trans,
16
+ Attr attr,
17
+ const std::vector<sycl::event>& deps = {});
18
+
19
+ TORCH_API sycl::event convolution(
20
+ at::Tensor& dst,
21
+ const at::Tensor& src,
22
+ const at::Tensor& weight,
23
+ const at::Tensor& bia,
24
+ IntArrayRef padding_front_top_left,
25
+ IntArrayRef padding_back_bottom_right,
26
+ IntArrayRef stride,
27
+ IntArrayRef dilation,
28
+ int64_t groups,
29
+ Attr& attr,
30
+ const std::vector<sycl::event>& deps = {});
31
+
32
+ TORCH_API sycl::event convolution_backward_weights(
33
+ at::Tensor& diff_weight,
34
+ at::Tensor& diff_bia,
35
+ const at::Tensor& diff_dst,
36
+ const at::Tensor& src,
37
+ IntArrayRef diff_weight_aten_size,
38
+ IntArrayRef padding_front_top_left,
39
+ IntArrayRef padding_back_bottom_right,
40
+ IntArrayRef stride,
41
+ IntArrayRef dilation,
42
+ int64_t groups,
43
+ const std::vector<sycl::event>& deps = {});
44
+
45
+ TORCH_API sycl::event convolution_backward_data(
46
+ at::Tensor& diff_src,
47
+ const at::Tensor& diff_dst,
48
+ const at::Tensor& weight,
49
+ IntArrayRef padding_front_top_left,
50
+ IntArrayRef padding_back_bottom_right,
51
+ IntArrayRef stride,
52
+ IntArrayRef dilation,
53
+ int64_t groups,
54
+ bool bias_defined,
55
+ const std::vector<sycl::event>& deps = {});
56
+
57
+ TORCH_API sycl::event deconvolution(
58
+ at::Tensor& dst,
59
+ const at::Tensor& src,
60
+ const at::Tensor& weight,
61
+ const at::Tensor& bia,
62
+ IntArrayRef stride,
63
+ IntArrayRef padding,
64
+ IntArrayRef dst_padding,
65
+ IntArrayRef dilation,
66
+ int64_t groups,
67
+ Attr& attr,
68
+ const std::vector<sycl::event>& deps = {});
69
+
70
+ TORCH_API sycl::event deconvolution_backward_data(
71
+ at::Tensor& diff_src,
72
+ const at::Tensor& diff_dst,
73
+ const at::Tensor& weight,
74
+ IntArrayRef stride,
75
+ IntArrayRef padding,
76
+ IntArrayRef dilation,
77
+ int64_t groups,
78
+ bool bias_defined,
79
+ const std::vector<sycl::event>& deps = {});
80
+
81
+ TORCH_API sycl::event deconvolution_backward_weights(
82
+ at::Tensor& diff_weight,
83
+ at::Tensor& diff_bia,
84
+ const at::Tensor& diff_dst,
85
+ const at::Tensor& src,
86
+ IntArrayRef stride,
87
+ IntArrayRef padding,
88
+ IntArrayRef dilation,
89
+ int64_t groups,
90
+ const std::vector<sycl::event>& deps = {});
91
+
92
+ dnnl::memory::dims conv_dst_size(
93
+ int64_t ndim,
94
+ IntArrayRef src_tz,
95
+ IntArrayRef wgh_tz,
96
+ IntArrayRef padding_front_top_left,
97
+ IntArrayRef padding_back_bottom_right,
98
+ IntArrayRef stride,
99
+ IntArrayRef dilation);
100
+
101
+ dnnl::memory::dims deconv_dst_size(
102
+ IntArrayRef src_size,
103
+ IntArrayRef wgh_size,
104
+ IntArrayRef padding,
105
+ IntArrayRef stride,
106
+ IntArrayRef dilation,
107
+ IntArrayRef dst_padding,
108
+ int64_t groups);
109
+
110
+ at::Tensor quantized_convolution(
111
+ at::Tensor act,
112
+ double act_scale,
113
+ int64_t act_zero_point,
114
+ at::Tensor weight,
115
+ at::Tensor weight_scales,
116
+ at::Tensor weight_zero_points,
117
+ c10::optional<at::Tensor> bias,
118
+ torch::List<int64_t> stride,
119
+ torch::List<int64_t> padding,
120
+ torch::List<int64_t> dilation,
121
+ bool transposed,
122
+ int64_t groups,
123
+ at::Tensor output,
124
+ double inv_output_scale,
125
+ int64_t output_zero_point,
126
+ c10::optional<at::Tensor> accum,
127
+ double accum_scale,
128
+ int64_t accum_zero_point,
129
+ c10::optional<c10::ScalarType> output_dtype,
130
+ c10::optional<std::string_view> binary_attr,
131
+ c10::optional<at::Scalar> binary_alpha,
132
+ c10::optional<std::string_view> unary_attr,
133
+ torch::List<c10::optional<at::Scalar>> unary_scalars,
134
+ c10::optional<std::string_view> unary_algorithm);
135
+
136
+ } // namespace at::native::onednn