| Klement Sekera | 558ceab | 2021-04-08 19:37:41 +0200 | [diff] [blame] | 1 | import os |
| 2 | import psutil |
| 3 | |
| 4 | available_cpus = psutil.Process().cpu_affinity() |
| 5 | num_cpus = len(available_cpus) |
| 6 | |
| 7 | max_vpp_cpus = os.getenv("MAX_VPP_CPUS", "auto").lower() |
| 8 | |
| 9 | if max_vpp_cpus == "auto": |
| 10 | max_vpp_cpus = num_cpus |
| 11 | else: |
| 12 | try: |
| 13 | max_vpp_cpus = int(max_vpp_cpus) |
| 14 | except ValueError as e: |
| 15 | raise ValueError("Invalid MAX_VPP_CPUS value specified, valid " |
| 16 | "values are a positive integer or 'auto'") from e |
| 17 | if max_vpp_cpus <= 0: |
| 18 | raise ValueError("Invalid MAX_VPP_CPUS value specified, valid " |
| 19 | "values are a positive integer or 'auto'") |
| 20 | if max_vpp_cpus > num_cpus: |
| 21 | max_vpp_cpus = num_cpus |