提交 689a5995 authored 作者: Frederic's avatar Frederic

Move code to c_code_support.

上级 987f2016
...@@ -3140,7 +3140,7 @@ class GpuEye(GpuOp): ...@@ -3140,7 +3140,7 @@ class GpuEye(GpuOp):
def __hash__(self): def __hash__(self):
return hash(self.dtype) ^ hash(type(self)) return hash(self.dtype) ^ hash(type(self))
def c_support_code_apply(self, node, nodename): def c_support_code(self):
return """ return """
//Only 1 block is used. //Only 1 block is used.
__global__ void kEye(float* a, int n, int m) { __global__ void kEye(float* a, int n, int m) {
...@@ -3196,5 +3196,5 @@ __global__ void kEye(float* a, int n, int m) { ...@@ -3196,5 +3196,5 @@ __global__ void kEye(float* a, int n, int m) {
return s return s
def c_code_cache_version(self): def c_code_cache_version(self):
return (2,) return (3,)
gpu_eye = GpuEye(dtype='float32') gpu_eye = GpuEye(dtype='float32')
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论