| 1234567891011121314151617181920212223 |
- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- #pragma once
- #include <ATen/core/Tensor.h>
- #include <ATen/miopen/miopen-wrapper.h>
- #include <ATen/miopen/Handle.h>
- namespace at { namespace native {
- // This function makes tensors which have zero stride contiguous, by
- // setting the strides to 1.
- inline Tensor contiguousIfZeroInStrides(const Tensor& t) {
- for (auto s : t.strides()) {
- if (s == 0) return t.contiguous();
- }
- return t;
- }
- }}
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
