提交 994f2486 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add an import for os and fix some flake8 warnings.

上级 c6b765d5
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import sys import sys
import os
import logging import logging
import sys
import warnings import warnings
import theano import theano
...@@ -31,6 +31,7 @@ from .type import (GpuArrayType, GpuArrayVariable, GpuArrayConstant, ...@@ -31,6 +31,7 @@ from .type import (GpuArrayType, GpuArrayVariable, GpuArrayConstant,
from .basic_ops import as_gpuarray_variable from .basic_ops import as_gpuarray_variable
from . import fft, dnn, opt, nerv, extra_ops, multinomial, reduction from . import fft, dnn, opt, nerv, extra_ops, multinomial, reduction
def transfer(x, target): def transfer(x, target):
try: try:
get_context(target) get_context(target)
...@@ -94,8 +95,8 @@ def init_dev(dev, name=None, preallocate=None): ...@@ -94,8 +95,8 @@ def init_dev(dev, name=None, preallocate=None):
pygpu.empty((gmem,), dtype='int8', context=context) pygpu.empty((gmem,), dtype='int8', context=context)
if config.print_active_device: if config.print_active_device:
print("Preallocating %d/%d Mb (%f) on %s" % print("Preallocating %d/%d Mb (%f) on %s" %
(gmem//MB, context.total_gmem//MB, (gmem // MB, context.total_gmem // MB,
gmem/context.total_gmem, dev), gmem / context.total_gmem, dev),
file=sys.stderr) file=sys.stderr)
# Initialise the blas kernels. We do this after the # Initialise the blas kernels. We do this after the
...@@ -170,7 +171,7 @@ def use(device, ...@@ -170,7 +171,7 @@ def use(device,
if pygpu: if pygpu:
try: try:
if (config.device.startswith('cuda') or if (config.device.startswith('cuda') or
config.device.startswith('opencl')): config.device.startswith('opencl')):
use(config.device) use(config.device)
elif (config.init_gpu_device.startswith('cuda') or elif (config.init_gpu_device.startswith('cuda') or
config.init_gpu_device.startswith('opencl')): config.init_gpu_device.startswith('opencl')):
...@@ -191,8 +192,7 @@ if pygpu: ...@@ -191,8 +192,7 @@ if pygpu:
from .basic_ops import (GpuAlloc, GpuAllocEmpty, GpuContiguous, GpuEye, from .basic_ops import (GpuAlloc, GpuAllocEmpty, GpuContiguous, GpuEye,
GpuFromHost, GpuJoin, GpuReshape, GpuSplit, GpuFromHost, GpuJoin, GpuReshape, GpuSplit,
HostFromGpu) HostFromGpu, host_from_gpu)
from .basic_ops import host_from_gpu, GpuFromHost
from .elemwise import GpuElemwise from .elemwise import GpuElemwise
from .subtensor import (GpuSubtensor, GpuIncSubtensor, from .subtensor import (GpuSubtensor, GpuIncSubtensor,
GpuAdvancedIncSubtensor1) GpuAdvancedIncSubtensor1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论