alihan/exo
1
0
Fork 0
mirror of https://github.com/exo-explore/exo.git synced 2025-10-23 02:57:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

better bench system info

Browse Source
This commit is contained in:
Alex Cheema
2024-12-12 11:34:37 +00:00
parent 8c6d37d9b8
commit f9f76129a1
1 changed files with 101 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

124
.github/bench.py vendored
View File

@@ -17,15 +17,15 @@ def check_system_state():
# Add macOS-specific checks # Add macOS-specific checks
try: try:
# Check powermetrics # Check powermetrics with sudo
try: try:
power_metrics = subprocess.run(['powermetrics', '-n', '1', '-i', '1000', '--show-process-energy'], power_metrics = subprocess.run(
capture_output=True, text=True) ['sudo', 'powermetrics', '-n', '1', '-i', '1000', '--samplers', 'cpu_power'],
except: capture_output=True, text=True
# Try with sudo if direct access fails )
power_metrics = subprocess.run(['sudo', 'powermetrics', '-n', '1', '-i', '1000', '--show-process-energy'], print("\nPower Metrics:", power_metrics.stdout, flush=True)
capture_output=True, text=True) except Exception as e:
print("\nPower Metrics:", power_metrics.stdout, flush=True) print(f"Error getting power metrics: {e}", flush=True)
# Check thermal state # Check thermal state
thermal_state = subprocess.run(['pmset', '-g', 'therm'], capture_output=True, text=True) thermal_state = subprocess.run(['pmset', '-g', 'therm'], capture_output=True, text=True)
@@ -35,9 +35,17 @@ def check_system_state():
arch = subprocess.run(['arch'], capture_output=True, text=True) arch = subprocess.run(['arch'], capture_output=True, text=True)
print("\nArchitecture:", arch.stdout, flush=True) print("\nArchitecture:", arch.stdout, flush=True)
# Check MLX compilation mode # Check MLX compilation mode - only if mlx is available
import mlx.core as mx try:
print("\nMLX Build Info:", mx.build_info(), flush=True) import mlx.core as mx
if hasattr(mx, 'build_info'):
print("\nMLX Build Info:", mx.build_info(), flush=True)
else:
print("\nMLX Build Info: Not available in this version", flush=True)
except ImportError:
print("\nMLX: Not installed", flush=True)
except Exception as e:
print(f"\nError checking MLX: {e}", flush=True)
except Exception as e: except Exception as e:
print(f"Error in macOS checks: {e}", flush=True) print(f"Error in macOS checks: {e}", flush=True)
@@ -45,13 +53,77 @@ def check_system_state():
# CPU Info # CPU Info
print("\nCPU Information:", flush=True) print("\nCPU Information:", flush=True)
try: try:
cpu_freq = psutil.cpu_freq() if platform.system() == 'Darwin' and platform.processor() == 'arm':
print(f"CPU Frequency - Current: {cpu_freq.current:.2f}MHz, Min: {cpu_freq.min:.2f}MHz, Max: {cpu_freq.max:.2f}MHz", flush=True) # Use sysctl for Apple Silicon Macs
print(f"CPU Usage per Core: {psutil.cpu_percent(percpu=True)}%", flush=True) cpu_info = subprocess.run(['sysctl', 'machdep.cpu'], capture_output=True, text=True)
if cpu_info.returncode == 0:
print(f"CPU Info (Apple Silicon):", cpu_info.stdout, flush=True)
# Parse powermetrics output for clearer CPU frequency display
try:
power_metrics = subprocess.run(
['sudo', 'powermetrics', '-n', '1', '-i', '100', '--samplers', 'cpu_power'],
capture_output=True, text=True
)
if power_metrics.returncode == 0:
output = power_metrics.stdout
print("\nDetailed CPU Frequency Information:")
# Extract cluster frequencies and max frequencies
current_cluster = None
max_freqs = {'E': 0, 'P0': 0, 'P1': 0}
for line in output.split('\n'):
# Track which cluster we're processing
if "E-Cluster" in line:
current_cluster = 'E'
elif "P0-Cluster" in line:
current_cluster = 'P0'
elif "P1-Cluster" in line:
current_cluster = 'P1'
# Get current frequencies
if "HW active frequency:" in line:
freq = line.split(':')[1].strip()
if freq != "0 MHz":
print(f"Current {current_cluster}-Cluster Frequency: {freq}")
# Get max frequencies from residency lines
if current_cluster and "active residency:" in line and "MHz:" in line:
try:
# Extract all frequency values
freqs = []
parts = line.split('MHz:')[:-1] # Skip last part as it's not a frequency
for part in parts:
freq_str = part.split()[-1]
try:
freq = float(freq_str)
freqs.append(freq)
except ValueError:
continue
if freqs:
max_freqs[current_cluster] = max(max_freqs[current_cluster], max(freqs))
except Exception:
continue
# Print max frequencies
print("\nMaximum Available Frequencies:")
for cluster, max_freq in max_freqs.items():
if max_freq > 0:
print(f"{cluster}-Cluster Max: {max_freq:.0f} MHz")
except Exception as e:
print(f"Error parsing powermetrics: {e}", flush=True)
else:
# Use psutil for other systems
cpu_freq = psutil.cpu_freq()
print(f"CPU Frequency - Current: {cpu_freq.current:.2f}MHz, Min: {cpu_freq.min:.2f}MHz, Max: {cpu_freq.max:.2f}MHz", flush=True)
print(f"\nCPU Usage per Core: {psutil.cpu_percent(percpu=True)}%", flush=True)
# Check if running in low power mode # Check if running in low power mode
power_mode = subprocess.run(['pmset', '-g'], capture_output=True, text=True) power_mode = subprocess.run(['pmset', '-g'], capture_output=True, text=True)
print("Power Settings:", power_mode.stdout, flush=True) print("\nPower Settings:", power_mode.stdout, flush=True)
except Exception as e: except Exception as e:
print(f"Error getting CPU info: {e}", flush=True) print(f"Error getting CPU info: {e}", flush=True)
@@ -88,27 +160,33 @@ def check_system_state():
try: try:
current_process = psutil.Process() current_process = psutil.Process()
print(f"Process Nice Value: {current_process.nice()}", flush=True) print(f"Process Nice Value: {current_process.nice()}", flush=True)
print(f"Process IO Nice Value: {current_process.ionice()}", flush=True) # Only try to get ionice if the platform supports it
print(f"Process CPU Affinity: {current_process.cpu_affinity()}", flush=True) if hasattr(current_process, 'ionice'):
print(f"Process IO Nice Value: {current_process.ionice()}", flush=True)
except Exception as e: except Exception as e:
print(f"Error getting process priority info: {e}", flush=True) print(f"Error getting process priority info: {e}", flush=True)
# System Load # System Load
print("\nSystem Load:", flush=True) print("\nSystem Load:", flush=True)
try: try:
print(f"Load Average: {psutil.getloadavg()}", flush=True) load_avg = psutil.getloadavg()
print(f"Load Average: {load_avg}", flush=True)
# Get top processes by CPU and Memory # Get top processes by CPU and Memory
print("\nTop Processes:", flush=True) print("\nTop Processes by CPU Usage:", flush=True)
processes = [] processes = []
for proc in psutil.process_iter(['pid', 'name', 'cpu_percent', 'memory_percent']): for proc in psutil.process_iter(['pid', 'name', 'cpu_percent', 'memory_percent']):
try: try:
processes.append(proc.info) pinfo = proc.info
if pinfo['cpu_percent'] is not None and pinfo['memory_percent'] is not None:
processes.append(pinfo)
except (psutil.NoSuchProcess, psutil.AccessDenied): except (psutil.NoSuchProcess, psutil.AccessDenied):
pass continue
sorted_by_cpu = sorted(processes, key=lambda x: x['cpu_percent'], reverse=True)[:5] # Sort and display top 5 CPU-consuming processes
print("Top 5 CPU-consuming processes:", json.dumps(sorted_by_cpu, indent=2), flush=True) sorted_by_cpu = sorted(processes, key=lambda x: x['cpu_percent'] or 0, reverse=True)[:5]
for proc in sorted_by_cpu:
print(f"PID: {proc['pid']}, Name: {proc['name']}, CPU: {proc['cpu_percent']}%, Memory: {proc['memory_percent']:.1f}%")
except Exception as e: except Exception as e:
print(f"Error getting system load info: {e}", flush=True) print(f"Error getting system load info: {e}", flush=True)