HeterogeneousCore/CUDAUtilities/scripts/cudaKernelStackBudget - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/scripts/cudaKernelStackBudget

12 строк387 B

Luca Ferragina

Add cudaKernelStackBudget tool to estimate CUDA stack-frame memory budget

13 июл 2026, 10:54

13 июл 2026, 10:54a05e6de

#!/usr/bin/env python3
"""Estimate the worst-case CUDA local-memory backing store reserved by a cmsRun config.

Thin command-line wrapper; the implementation lives in
HeterogeneousCore.CUDAUtilities.kernelStackBudget so it can be unit tested.
"""
import sys

from HeterogeneousCore.CUDAUtilities.kernelStackBudget import main

if __name__ == "__main__":
    sys.exit(main(sys.argv[1:]))