提交 88d99415 authored 作者: Abhinav-Khot's avatar Abhinav-Khot 提交者: Ricardo Vieira

Add support for negative values in psi

上级 88476b4f
......@@ -378,6 +378,13 @@ class Psi(UnaryScalarOp):
def c_support_code(self, **kwargs):
return """
// For GPU support
#ifdef WITHIN_KERNEL
#define DEVICE WITHIN_KERNEL
#else
#define DEVICE
#endif
#ifndef M_PI
#define M_PI 3.14159265358979323846
#endif
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论