Module enumerate 

SM-aware engine plan enumeration and counting.

Functions

count_engine_files 🔒

Counts every .engine file in dir regardless of _smXX suffix.

count_engine_files_for_sm 🔒

Counts .engine files in dir that match the requested SM.

engine_basenames_for_sm 🔒

Returns sorted basenames of .engine files under engine_dir that match the requested SM, or all .engine basenames when sm is None.

engine_basenames_in_dir_sorted 🔒

Returns sorted basenames of every .engine file under engine_dir, regardless of _smXX suffix.

engine_files_for_sm 🔒

Returns full paths of .engine files under engine_dir that match the requested SM, or all .engine files when sm is None.

matches_sm_suffix 🔒

Returns true when name is a TRT engine plan basename whose _smXX suffix matches the requested SM exactly.