diff --git a/include/rmm/device_buffer.hpp b/include/rmm/device_buffer.hpp index 4c700bce4..167d07c4b 100644 --- a/include/rmm/device_buffer.hpp +++ b/include/rmm/device_buffer.hpp @@ -109,7 +109,7 @@ class device_buffer { */ explicit device_buffer(std::size_t size, cuda_stream_view stream, - rmm::device_async_resource_ref mr = mr::get_current_device_resource()) + device_async_resource_ref mr = mr::get_current_device_resource()) : _stream{stream}, _mr{mr} { cuda_set_device_raii dev{_device}; @@ -138,7 +138,7 @@ class device_buffer { device_buffer(void const* source_data, std::size_t size, cuda_stream_view stream, - rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource()) + device_async_resource_ref mr = mr::get_current_device_resource()) : _stream{stream}, _mr{mr} { cuda_set_device_raii dev{_device}; @@ -169,7 +169,7 @@ class device_buffer { */ device_buffer(device_buffer const& other, cuda_stream_view stream, - rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource()) + device_async_resource_ref mr = mr::get_current_device_resource()) : device_buffer{other.data(), other.size(), stream, mr} { } diff --git a/include/rmm/device_uvector.hpp b/include/rmm/device_uvector.hpp index 100a70839..0750ef796 100644 --- a/include/rmm/device_uvector.hpp +++ b/include/rmm/device_uvector.hpp @@ -123,10 +123,9 @@ class device_uvector { * @param stream The stream on which to perform the allocation * @param mr The resource used to allocate the device storage */ - explicit device_uvector( - std::size_t size, - cuda_stream_view stream, - rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource()) + explicit device_uvector(std::size_t size, + cuda_stream_view stream, + device_async_resource_ref mr = mr::get_current_device_resource()) : _storage{elements_to_bytes(size), stream, mr} { } @@ -140,10 +139,9 @@ class device_uvector { * @param stream The stream on which to perform the copy * @param mr The resource used to allocate device memory for the new vector */ - explicit device_uvector( - device_uvector const& other, - cuda_stream_view stream, - rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource()) + explicit device_uvector(device_uvector const& other, + cuda_stream_view stream, + device_async_resource_ref mr = mr::get_current_device_resource()) : _storage{other._storage, stream, mr} { }