Skip to content

Commit cec1944

Browse files
authored
docs: add bfloat16 option to the model precision choice (#4866)
The parameter "precision" supports bfloat16 with most backends, but it is not documented. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **New Features** * Added support for "bfloat16" as a selectable precision option. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
1 parent 813bbc8 commit cec1944

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
"select_idx_map",
4545
]
4646

47-
_PRECISION = Literal["default", "float16", "float32", "float64"]
47+
_PRECISION = Literal["default", "float16", "bfloat16", "float32", "float64"]
4848
_ACTIVATION = Literal[
4949
"relu",
5050
"relu6",

0 commit comments

Comments
 (0)