找不到新添加Operator的 GPU 实现?

实现了一个新简单Op,CPU端的前向、后向计算是正常的,但把输入数据放到GPU上然后做计算,就会报错:

mxnet.base.MXNetError: Traceback (most recent call last):
  File "../src/imperative/imperative.cc", line 90
MXNetError: Operator CustomPooling is not implemented for GPU.

然后我在CUDA代码里面加一个语法错误,然后编译并没有报错,看样子这个.cu文件应该是没有被编译。但按照CMakeLists.txt里面的写法,应该自动把我的.cu文件加入到编译啊。

不知道问题出在哪啊?

重新跑一下cmake -GNinja .. 貌似就可以了。。。