Prompt48 commited on
Commit
ffd966f
·
verified ·
1 Parent(s): 20fe56f

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

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//cuda//Distributions.h ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ namespace at {
4
+ struct CUDAGeneratorImpl;
5
+ struct TensorIteratorBase;
6
+ class TensorBase;
7
+
8
+ namespace native {
9
+
10
+ void launch_poisson_cuda_kernel(
11
+ const TensorBase &ret, const TensorBase &lambda, CUDAGeneratorImpl *gen);
12
+
13
+ void launch_gamma_kernel(
14
+ const TensorBase &ret, const TensorBase &alpha, CUDAGeneratorImpl *gen);
15
+
16
+ void launch_binomial_cuda_kernel(
17
+ TensorIteratorBase &iter, CUDAGeneratorImpl *gen);
18
+
19
+ void launch_dirichlet_kernel(TensorIteratorBase &iter);
20
+
21
+ void launch_standard_gamma_grad_kernel(TensorIteratorBase &iter);
22
+
23
+ void launch_dirichlet_grad_kernel(TensorIteratorBase &iter);
24
+
25
+ }} // namespace at::native