Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 3e565eb

Browse files
committed
Add bf16 to runner_et test
ghstack-source-id: 281d80f ghstack-comment-id: 2372312701 Pull Request resolved: #1196
1 parent d6954a9 commit 3e565eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/pull.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -941,7 +941,7 @@ jobs:
941941
python torchchat.py export stories15M --output-pte-path ./model.pte
942942
./cmake-out/et_run ./model.pte -z ./tokenizer.model -t 0 -i "${PRMT}"
943943
944-
for dtype in fp32 fp16; do # bf16 needs to be supported
944+
for dtype in fp32 fp16 bf16; do
945945
echo "Testing export + runner with dtype=$dtype"
946946
python torchchat.py export stories15M --dtype $dtype --output-pte-path ./model.pte
947947
./cmake-out/et_run ./model.pte -z ./tokenizer.model -t 0 -i "${PRMT}"

0 commit comments

Comments
 (0)