Skip to content

Commit 1462e19

Browse files
fix(ci): fix missing load_profile calls and Windows build issues
- Added load_profile to all remaining benchmark and test steps - Safely handle ROCM_HOME on Windows during setup - Final synchronization for CI/CD
1 parent 904107d commit 1462e19

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

.github/workflows/production.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ jobs:
197197
198198
- name: Run Basic Test
199199
run: |
200-
python -c "from crayon import CrayonVocab; v = CrayonVocab(device='cpu'); print(v.tokenize('Hello Windows!'))"
200+
python -c "from crayon import CrayonVocab; v = CrayonVocab(device='cpu'); v.load_profile('lite'); print(v.tokenize('Hello Windows!'))"
201201
202202
# ==========================================================================
203203
# JOB 5: BENCHMARK (CPU Performance Validation)
@@ -261,6 +261,7 @@ jobs:
261261
from crayon import CrayonVocab
262262
263263
v = CrayonVocab(device='cpu')
264+
v.load_profile('lite')
264265
265266
# Generate test text
266267
text = 'The quick brown fox jumps over the lazy dog. ' * 10000

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,9 @@ def log(msg: str, level: str = "INFO") -> None:
6565

6666
# Detect ROCm
6767
FORCE_ROCM = os.environ.get("CRAYON_FORCE_ROCM", "0") == "1"
68-
ROCM_HOME = os.environ.get("ROCM_HOME", "/opt/rocm")
69-
HIPCC_PATH = os.path.join(ROCM_HOME, "bin", "hipcc")
70-
HAS_ROCM = (os.path.exists(HIPCC_PATH) or FORCE_ROCM)
68+
ROCM_HOME = os.environ.get("ROCM_HOME", "/opt/rocm" if sys.platform != "win32" else "")
69+
HIPCC_PATH = os.path.join(ROCM_HOME, "bin", "hipcc") if ROCM_HOME else ""
70+
HAS_ROCM = (os.path.exists(HIPCC_PATH) if HIPCC_PATH else False) or FORCE_ROCM
7171

7272
if HAS_ROCM:
7373
log(f"ROCm detected at {ROCM_HOME} (Forced={FORCE_ROCM})")

0 commit comments

Comments
 (0)