thrust::device_malloc使用方法

`thrust::device_malloc`是Thrust库提供的一种内存分配函数，用于在GPU上分配内存空间。它的使用方法与`cudaMalloc`类似，但具有更好的错误处理和异常抛出机制。以下是一个示例代码：

#include <thrust/device_malloc.h>
#include <thrust/device_ptr.h>

int main()
{
    int n = 1000;

    // 申请n个int类型的空间
    thrust::device_ptr<int> d_vec(thrust::device_malloc<int>(n));

    // 申请n个float类型的空间
    thrust::device_ptr<float> d_floats(thrust::device_malloc<float>(n));

    // 申请n个bool类型的空间
    thrust::device_ptr<bool> d_bools(thrust::device_malloc<bool>(n));

    // 使用完后必须手动释放内存空间
    thrust::device_free(d_vec);
    thrust::device_free(d_floats);
    thrust::device_free(d_bools);

    return 0;
}

在这个示例中，我们使用`thrust::device_malloc`来申请内存空间，并通过`thrust::device_ptr`将其封装成一个指针。使用完毕后，我们需要手动使用`thrust::device_free`函数释放内存空间。

需要注意的是，使用`thrust::device_malloc`和`thrust::device_free`进行内存管理时，需要确保在内存释放之前，所有对内存空间的引用都已被释放。否则，可能会导致内存泄漏或者使用已释放的内存空间，从而引发不可预期的结果。