Prompt48 commited on
Commit
bb53c92
·
verified ·
1 Parent(s): 40e5145

Upload edit\Qwen3-TTS-test\.venv\Lib\site-packages\torch\include\ATen\native\cuda\SortStable.h with huggingface_hub

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//cuda//SortStable.h ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+ #include <ATen/core/TensorBase.h>
3
+ #include <cstdint>
4
+
5
+ namespace at::native {
6
+
7
+ // Stable-sort self into values, and set indices to the
8
+ // inverse-permutation from values back to self.
9
+ // Output tensors must be pre-allocated and contiguous.
10
+ void launch_stable_sort_kernel(
11
+ const TensorBase& self,
12
+ int64_t dim,
13
+ bool descending,
14
+ const TensorBase& values,
15
+ const TensorBase& indices);
16
+
17
+ } // namespace at::native