Skip to content

Commit

Permalink
Fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
gmarkall committed Mar 4, 2024
1 parent 5fb859f commit a3a32a7
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions pynvjitlink/tests/test_pynvjitlink.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,8 +146,9 @@ def test_get_linked_cubin(device_functions_cubin, gpu_arch_flag):
assert cubin[:4] == b"\x7fELF"


def test_get_linked_cubin_link_not_complete_error(device_functions_cubin,
gpu_arch_flag):
def test_get_linked_cubin_link_not_complete_error(
device_functions_cubin, gpu_arch_flag
):
handle = _nvjitlinklib.create(gpu_arch_flag)
filename, data = device_functions_cubin
input_type = InputType.CUBIN.value
Expand Down Expand Up @@ -188,8 +189,7 @@ def test_get_linked_ptx_from_lto(device_functions_ltoir, gpu_arch_flag):
_nvjitlinklib.destroy(handle)


def test_get_linked_ptx_link_not_complete_error(device_functions_ltoir,
gpu_arch_flag):
def test_get_linked_ptx_link_not_complete_error(device_functions_ltoir, gpu_arch_flag):
handle = _nvjitlinklib.create(gpu_arch_flag, "-lto", "-ptx")
filename, data = device_functions_ltoir
input_type = InputType.OBJECT.value
Expand Down

0 comments on commit a3a32a7

Please sign in to comment.