diff --git a/src/backend/cuda/ThrustArrayFirePolicy.hpp b/src/backend/cuda/ThrustArrayFirePolicy.hpp index 189ee558b3..339d3ea088 100644 --- a/src/backend/cuda/ThrustArrayFirePolicy.hpp +++ b/src/backend/cuda/ThrustArrayFirePolicy.hpp @@ -37,7 +37,11 @@ inline void return_temporary_buffer(ThrustArrayFirePolicy, Pointer p) { } // namespace cuda } // namespace arrayfire +#if defined(_WIN32) +THRUST_NAMESPACE_BEGIN +#else namespace thrust { +#endif namespace cuda_cub { template<> __DH__ inline cudaStream_t get_stream( @@ -60,4 +64,8 @@ inline cudaError_t synchronize_stream( } } // namespace cuda_cub +#if defined(_WIN32) +THRUST_NAMESPACE_END +#else } // namespace thrust +#endif