1from __future__ import absolute_import, division, print_function, unicode_literals
2
3import torch_glow
4
5
6def pytest_addoption(parser):
7    parser.addoption("--backend", action="store", default=None)
8
9
10def pytest_sessionstart(session):
11    backend = session.config.getoption("--backend")
12    if backend:
13        torch_glow.setGlowBackend(backend)
14