| 137 | namespace cvcudapy { |
| 138 | |
| 139 | float4 GetFloat4FromPyArray(const pyarray &array) |
| 140 | { |
| 141 | if (array.ndim() == 0) |
| 142 | { |
| 143 | return nvcv::cuda::SetAll<float4>(*array.data()); |
| 144 | } |
| 145 | else |
| 146 | { |
| 147 | float4 value = nvcv::cuda::SetAll<float4>(0.f); |
| 148 | |
| 149 | for (int i = 0; i < static_cast<int>(array.size()); i++) |
| 150 | { |
| 151 | nvcv::cuda::GetElement(value, i) = *array.data(i); |
| 152 | } |
| 153 | |
| 154 | return value; |
| 155 | } |
| 156 | } |
| 157 | |
| 158 | } // namespace cvcudapy |