Skip to content

Commit

Permalink
Fix docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
wence- committed Dec 14, 2023
1 parent eb8f107 commit 1c426b3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/rmm/_lib/_torch_allocator.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
* @param size The number of bytes to allocate
* @param device The device whose memory resource one should use
* @param stream CUDA stream to perform allocation on
* @return void* Pointer to the newly allocated memory
* @return Pointer to the newly allocated memory
*/
extern "C" void* allocate(std::size_t size, int device, void* stream)
{
Expand Down

0 comments on commit 1c426b3

Please sign in to comment.