File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed
Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change @@ -65,9 +65,9 @@ def log(msg: str, level: str = "INFO") -> None:
6565
6666# Detect ROCm
6767FORCE_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
7272if HAS_ROCM :
7373 log (f"ROCm detected at { ROCM_HOME } (Forced={ FORCE_ROCM } )" )
You can’t perform that action at this time.
0 commit comments