Skip to content

Commit

Permalink
Update paddle.js
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Jun 25, 2024
1 parent cf0cc2f commit 6291550
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions source/paddle.js
Original file line number Diff line number Diff line change
Expand Up @@ -786,8 +786,11 @@ paddle.Utility = class {
const types = new Map([
['bool', 'boolean'],
['bf16', 'bfloat16'],
['fp16', 'float16'], ['fp32', 'float32'], ['fp64', 'float64'],
['fp8_e4m3fn', 'float8e4m3fn'], ['fp8_e5m2', 'float8e5m2']
['fp16', 'float16'],
['fp32', 'float32'],
['fp64', 'float64'],
['fp8_e4m3fn', 'float8e4m3fn'],
['fp8_e5m2', 'float8e5m2']
]);
for (const [name, index] of Object.entries(paddle.DataType)) {
const key = name.toLowerCase();
Expand Down Expand Up @@ -851,6 +854,10 @@ paddle.DataType = {
BF16: 22,
COMPLEX64: 23,
COMPLEX128: 24,
STRING: 25,
STRINGS: 26,
FP8_E4M3FN: 32,
FP8_E5M2: 33,
};

paddle.AttributeType = {
Expand Down

0 comments on commit 6291550

Please sign in to comment.