异常0xe06d7363，试图在libtorch中获取张量数据指针时

Question

我正在努力写一个CUDA操作员。数据的地址指针应传递到操作员。但是我遇到了这个例外，因为你会吼叫。

异常被像以下代码一样抛出：

#include <iostream>
#include <torch/serialize/tensor.h>
#include <ATen/cuda/CUDAContext.h>

int main() {
    // note that whether on gpu or cpu, the exception is the same
    auto options = at::TensorOptions().dtype(torch::kInt64).device(torch::kCUDA, 0); 
    auto x = at::arange(0, 10, options); 
    
    // get the pointer of data
    int* x_ptr = x.data_ptr<int>(); // where the exception is thrown out
    std::cout << "x[0]: " << x_ptr[0] << std::endl; // print out 
    
    // ....
    // pass x_ptr into operator function
    // ...
}

在CPU上，异常：异常0xE06D7363在地址遇到的0x7ffd416f4f4f4f6

在CUDA上，例外0xE06D7363在地址0x7ffd416f4f4f4f69

：win10

Platporl> Platform> Platporl compiator：win fins：win fins. 2022 CL版本19.31.31107