Skip to content

Update bitaxe_hashrate_benchmark.py #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Update bitaxe_hashrate_benchmark.py #3

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
484f186
Update bitaxe_hashrate_benchmark.py
Co1nB3e Jun 17, 2025
f7d6d6d
Update bitaxe_hashrate_benchmark.py
Co1nB3e Jun 18, 2025
ea3a2b3
Update bitaxe_hashrate_benchmark.py
Co1nB3e Jun 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
65 changes: 57 additions & 8 deletions bitaxe_hashrate_benchmark.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,19 @@
import sys
import argparse

# Enable ansi escape characters in terminal - colors were not working in Windows terminal
import os
try:
import colorama
colorama.init()
except ImportError:
# Fallback for environments where colorama isn't available
if os.name == "nt":
os.system("") # rudimentary ANSI enable on Windows

# Compute timestamp for file suffix
timestamp = time.strftime("%Y%m%d-%H%M%S")

# ANSI Color Codes
GREEN = "\033[92m"
YELLOW = "\033[93m"
Expand Down Expand Up @@ -72,6 +85,19 @@ def parse_arguments():
if benchmark_time / sample_interval < 7:
raise ValueError(RED + f"Error: Benchmark time is too short. Please increase the benchmark time or decrease the sample interval. At least 7 samples are required." + RESET)

# Add suffix to filename in case of manual initial voltage/frequency
file_suffix = ""
if initial_voltage != 1150:
file_suffix = file_suffix + "_v" + str(initial_voltage)
if initial_frequency != 500:
file_suffix = file_suffix + "_f" + str(initial_frequency)

# Refactor filename (called in multiple places)
def result_filename():
# Extract IP from bitaxe_ip global variable and remove 'http://'
ip_address = bitaxe_ip.replace('http://', '')
return f"bitaxe_benchmark_results_{ip_address}_{timestamp}{file_suffix}.json"

# Results storage
results = []

Expand Down Expand Up @@ -305,9 +331,8 @@ def benchmark_iteration(core_voltage, frequency):

def save_results():
try:
# Extract IP from bitaxe_ip global variable and remove 'http://'
ip_address = bitaxe_ip.replace('http://', '')
filename = f"bitaxe_benchmark_results_{ip_address}.json"
# Refactored filename computation
filename = result_filename()
with open(filename, "w") as f:
json.dump(results, f, indent=4)
print(GREEN + f"Results saved to {filename}" + RESET)
Expand Down Expand Up @@ -346,12 +371,14 @@ def reset_to_best_setting():

current_voltage = initial_voltage
current_frequency = initial_frequency
retry_upon_overheat = 0

while current_voltage <= max_allowed_voltage and current_frequency <= max_allowed_frequency:
set_system_settings(current_voltage, current_frequency)
avg_hashrate, avg_temp, efficiency_jth, hashrate_ok, avg_vr_temp, error_reason = benchmark_iteration(current_voltage, current_frequency)

if avg_hashrate is not None and avg_temp is not None and efficiency_jth is not None:
retry_upon_overheat = 0
result = {
"coreVoltage": current_voltage,
"frequency": current_frequency,
Expand All @@ -370,20 +397,41 @@ def reset_to_best_setting():
# If hashrate is good, try increasing frequency
if current_frequency + frequency_increment <= max_allowed_frequency:
current_frequency += frequency_increment
print(GREEN + "Hashrate is good. Increasing frequency for next try." + RESET)
else:
print(GREEN + "Reached max frequency with good results. Stopping further testing." + RESET)
break # We've reached max frequency with good results
else:
# If hashrate is not good, go back one frequency step and increase voltage
if current_voltage + voltage_increment <= max_allowed_voltage:
current_voltage += voltage_increment
current_frequency -= frequency_increment # Go back to one frequency step and retry
print(YELLOW + f"Hashrate to low compared to expected. Decreasing frequency to {current_frequency}MHz and increasing voltage to {current_voltage}mV" + RESET)
print(YELLOW + f"Hashrate too low compared to expected. Decreasing frequency to {current_frequency}MHz and increasing voltage to {current_voltage}mV" + RESET)
else:
print(YELLOW + "Reached max voltage without good results. Stopping further testing." + RESET)
break # We've reached max voltage without good results
else:
# If we hit thermal limits or other issues, we've found the highest safe settings
print(GREEN + "Reached thermal or stability limits. Stopping further testing." + RESET)
break # Stop testing higher values
# In case of max Chip Temperature reached, continue loop to next voltage with decreased frequency
# Condition added to avoid successive overheat tries and reset to high initial frequency
overheat_retry_allowed = (
error_reason == "CHIP_TEMP_EXCEEDED"
and retry_upon_overheat < 1
and initial_frequency <= current_frequency + frequency_increment
)
if overheat_retry_allowed:
# If overheat, return to initial frequency while increasing voltage (considering max_allowed_voltage)
retry_upon_overheat += 1
if current_voltage + voltage_increment <= max_allowed_voltage:
current_frequency = initial_frequency
current_voltage += voltage_increment
print(GREEN + "Reached thermal limit for the current voltage/frequency. Switching to next voltage increment." + RESET)
else:
print(GREEN + "Reached thermal limit for the current voltage/frequency. Next voltage increment out of voltage limit. Stopping further testing." + RESET)
break # We've reached max voltage, can't increase voltage anymore
else:
print(GREEN + "Reached thermal or stability limits. Stopping further testing." + RESET)
break # Stop testing higher values

save_results()

Expand Down Expand Up @@ -444,8 +492,8 @@ def reset_to_best_setting():
}

# Save the final data to JSON
ip_address = bitaxe_ip.replace('http://', '')
filename = f"bitaxe_benchmark_results_{ip_address}.json"
# Refactored filename computation
filename = result_filename()
with open(filename, "w") as f:
json.dump(final_data, f, indent=4)

Expand Down Expand Up @@ -495,3 +543,4 @@ def cleanup_and_exit(reason=None):
print(RED + f"Benchmarking stopped: {reason}" + RESET)
print(GREEN + "Benchmarking completed." + RESET)
sys.exit(0)