Prompt48 commited on
Commit
9c7ad76
·
verified ·
1 Parent(s): c168512

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

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//cuda//IndexKernel.h ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+ #include <c10/core/ScalarType.h>
3
+ #include <cstdint>
4
+
5
+ namespace at {
6
+ struct TensorIteratorBase;
7
+ class TensorBase;
8
+ }
9
+
10
+ namespace at::native {
11
+ /// @param maskPrefixSum[in,out]
12
+ void launch_masked_scatter_kernel(
13
+ const TensorBase &self, const TensorBase &mask,
14
+ const TensorBase &maskPrefixSum, const TensorBase &source);
15
+ }