def rjit(request):
local = False
rjit = RemoteJIT(debug=not True, local=local)
if not local:
rjit.start_server(background=True)
request.addfinalizer(rjit.stop_server)
atexit.register(rjit.stop_server)
return rjit
def with_localjit(test_func):
ljit = RemoteJIT(local=True)
device = tuple(ljit.targets)[0]
target_info = ljit.targets[device]
def test_func_():
with target_info:
return test_func(ljit)
return test_func_
def ljit(request):
ljit = RemoteJIT(debug=not True, local=True)
return ljit
def rjit(request):
rjit = RemoteJIT(debug=True)
rjit.start_server(background=True)
request.addfinalizer(rjit.stop_server)
atexit.register(rjit.stop_server)
return rjit
def djit():
"""
Debug JIT: a RemoteJIT() which automatically uses tracing_allocator and
detects memory leaks
"""
return RemoteJIT(local=True, debug=True, use_tracing_allocator=True)