Prompt48 commited on
Commit
3a9059d
·
verified ·
1 Parent(s): e96ac4c

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

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//cuda//Resize.h ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+
3
+ #include <ATen/EmptyTensor.h>
4
+ #include <ATen/native/ResizeCommon.h>
5
+
6
+ #include <c10/cuda/CUDAGuard.h>
7
+
8
+ namespace at { namespace native {
9
+
10
+ TORCH_CUDA_CPP_API void resize_bytes_cuda(StorageImpl* storage, size_t size_bytes);
11
+
12
+ static inline void maybe_resize_storage_cuda(TensorImpl* self, size_t new_size_bytes) {
13
+ // It does not make sense to try to resize a storage
14
+ // to hold 0 elements, and this can break
15
+ // if storage_offset is positive but
16
+ // new_size is 0, so just bail in that case
17
+ // (same comment is in Resize.h)
18
+ if (self->numel() == 0) {
19
+ return;
20
+ }
21
+
22
+ const Storage &storage = self->unsafe_storage();
23
+ TORCH_CHECK(storage, "Tensor: invalid null storage");
24
+ if (new_size_bytes > storage.nbytes()) {
25
+ resize_bytes_cuda(storage.unsafeGetStorageImpl(), new_size_bytes);
26
+ }
27
+ }
28
+
29
+ inline TensorImpl* resize_impl_cuda_(
30
+ TensorImpl* self,
31
+ IntArrayRef size,
32
+ at::OptionalIntArrayRef stride) {
33
+ if (self->sizes() == size && (!stride || self->strides() == stride)) {
34
+ return self;
35
+ }
36
+ const auto itemsize = self->dtype().itemsize();
37
+ const auto storage_offset = self->storage_offset();
38
+ size_t storage_size = 1;
39
+ if (stride) {
40
+ self->set_sizes_and_strides(size, *stride);
41
+ storage_size = at::detail::computeStorageNbytes(
42
+ size, *stride, itemsize, storage_offset);
43
+ } else {
44
+ self->set_sizes_contiguous(size);
45
+ storage_size = at::detail::computeStorageNbytesContiguous(
46
+ size, itemsize, storage_offset);
47
+ }
48
+ maybe_resize_storage_cuda(self, storage_size);
49
+
50
+ return self;
51
+ }
52
+
53
+ }}