Remove debug messages

这个提交包含在:
craig-warren
2023-12-10 22:30:11 +00:00
父节点 ab7252feb7
当前提交 107d2d2e6f
共有 2 个文件被更改,包括 1 次插入6 次删除

查看文件

@@ -352,7 +352,6 @@ class ModelBuildRun:
logger.info(
f"Time taken: " + f"{humanize.precisedelta(datetime.timedelta(seconds=solver.solvetime), format='%0.4f')}"
)
logger.basic(subprocess.check_output(["nvidia-smi"], shell=False, stderr=subprocess.STDOUT))
class GridBuilder:

查看文件

@@ -235,8 +235,6 @@ class CUDAUpdates:
# Import PyCUDA modules
self.drv = import_module("pycuda.driver")
self.clear_context_caches = getattr(import_module("pycuda.tools"),
"clear_context_caches")
self.source_module = getattr(import_module("pycuda.compiler"), "SourceModule")
self.drv.init()
@@ -786,11 +784,9 @@ class CUDAUpdates:
def cleanup(self):
"""Cleanup GPU context."""
# Remove context from top of stack and delete
# Remove context from top of stack and clear
self.ctx.pop()
self.ctx = None
self.clear_context_caches()
class OpenCLUpdates: