nvcc problem

customop.py

import torch
import torch.utils.cpp_extension

def compile():
    ext_name = 'ext1'
    torch.utils.cpp_extension.load(name=ext_name, sources=['ext1.cpp', 'ext1cuda.cu'], with_cuda=True, verbose=True)

def main():
    compile()

if __name__ == "__main__":
    main()

ext1.cpp

#include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAUtils.h>
#include <pybind11/numpy.h>

#include "op.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
}

ext1cuda.cu

#include "op.h"

void testfunc() {
    Context ctx;
    (void)ctx;
}

op.h

#pragma once

typedef struct {
    int foo;
} Context;