Unverified Commit dbc3d999 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[UX] Put CUDA attention backend selection log into one line (#29337)


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 794029f0
...@@ -407,9 +407,6 @@ class CudaPlatformBase(Platform): ...@@ -407,9 +407,6 @@ class CudaPlatformBase(Platform):
# We have found some valid backends. Select the one with the # We have found some valid backends. Select the one with the
# highest priority. # highest priority.
logger.info(
"Valid backends: %s", [b[0].name for b in valid_backends_priorities]
)
sorted_indices = sorted( sorted_indices = sorted(
range(len(valid_backends_priorities)), range(len(valid_backends_priorities)),
key=lambda i: valid_backends_priorities[i][1], key=lambda i: valid_backends_priorities[i][1],
...@@ -417,8 +414,9 @@ class CudaPlatformBase(Platform): ...@@ -417,8 +414,9 @@ class CudaPlatformBase(Platform):
selected_index = sorted_indices[0] selected_index = sorted_indices[0]
selected_backend = valid_backends_priorities[selected_index][0] selected_backend = valid_backends_priorities[selected_index][0]
logger.info( logger.info(
"Using %s backend.", "Using %s attention backend out of potential backends: %s",
selected_backend.name, selected_backend.name,
[b[0].name for b in valid_backends_priorities],
) )
return selected_backend.get_path() return selected_backend.get_path()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment