提交 5ffe17ad authored 作者: Ben Mares's avatar Ben Mares 提交者: Luciano Paz

Remove iostream include for C backend

上级 5229feba
...@@ -157,7 +157,7 @@ class DynamicModule: ...@@ -157,7 +157,7 @@ class DynamicModule:
self.support_code = [] self.support_code = []
self.functions = [] self.functions = []
self.includes = ["<Python.h>", "<iostream>", '"pytensor_mod_helper.h"'] self.includes = ["<Python.h>", '"pytensor_mod_helper.h"']
self.init_blocks = [] self.init_blocks = []
def print_methoddef(self, stream): def print_methoddef(self, stream):
......
...@@ -472,9 +472,7 @@ class GemmRelated(COp): ...@@ -472,9 +472,7 @@ class GemmRelated(COp):
return blas_header_text() + mod_str return blas_header_text() + mod_str
def c_headers(self, **kwargs): def c_headers(self, **kwargs):
# std.cout doesn't require the '%' symbol to print stuff... return []
# so it works much better with python's string-substitution stuff.
return ["<iostream>"]
def c_libraries(self, **kwargs): def c_libraries(self, **kwargs):
return ldflags() return ldflags()
......
...@@ -294,7 +294,7 @@ class Softmax(COp): ...@@ -294,7 +294,7 @@ class Softmax(COp):
return shape return shape
def c_headers(self, **kwargs): def c_headers(self, **kwargs):
return ["<iostream>", "<cmath>"] return ["<cmath>"]
def c_support_code_apply(self, node: Apply, name: str) -> str: def c_support_code_apply(self, node: Apply, name: str) -> str:
"""Needed to define NPY_RAVEL_AXIS""" """Needed to define NPY_RAVEL_AXIS"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论