def METHOD_NAME(session):
    """test function for gpu"""
    gpu_lib = None
    try:
        gpu_lib = export_gpu_add_lib()
        test_add(session, gpu_lib, "/gpu:0")
    finally:
        if gpu_lib is not None:
            os.remove(gpu_lib)