This commit is contained in:
Saeed Maleki
2023-10-23 04:25:37 +00:00
parent 842805fe68
commit 2afce036ad

View File

@@ -9,7 +9,15 @@ import numpy as np
import netifaces as ni
import pytest
from mscclpp import Fifo, Host2DeviceSemaphore, Host2HostSemaphore, ProxyService, SmDevice2DeviceSemaphore, Transport, get_ib_device_count
from mscclpp import (
Fifo,
Host2DeviceSemaphore,
Host2HostSemaphore,
ProxyService,
SmDevice2DeviceSemaphore,
Transport,
get_ib_device_count,
)
from ._cpp import _ext
from .mscclpp_group import MscclppGroup
from .mscclpp_mpi import MpiGroup, parametrize_mpi_groups, mpi_group
@@ -19,6 +27,7 @@ ethernet_interface_name = "eth0"
skipif_ib = pytest.mark.skipif(get_ib_device_count() == 0, reason="no IB device")
def parametrize_transport(*transports: list):
def decorator(func):
params = []
@@ -28,6 +37,7 @@ def parametrize_transport(*transports: list):
else:
params.append(transport)
return pytest.mark.parametrize("transport", params)(func)
return decorator