From b0a10fb1a30bfdbb8d23f9b19d5ceeb143ba3efd Mon Sep 17 00:00:00 2001 From: gururaj1512 Date: Mon, 22 Jun 2026 16:20:50 +0530 Subject: [PATCH] feat: add float16 dtype support to array/zeros --- .../@stdlib/array/zeros/README.md | 1 + .../array/zeros/benchmark/benchmark.js | 18 ++++ .../benchmark/benchmark.length.float16.js | 94 +++++++++++++++++++ .../@stdlib/array/zeros/docs/repl.txt | 1 + .../@stdlib/array/zeros/docs/types/index.d.ts | 1 + .../@stdlib/array/zeros/docs/types/test.ts | 1 + .../@stdlib/array/zeros/package.json | 4 + .../@stdlib/array/zeros/test/test.js | 15 +++ 8 files changed, 135 insertions(+) create mode 100644 lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.length.float16.js diff --git a/lib/node_modules/@stdlib/array/zeros/README.md b/lib/node_modules/@stdlib/array/zeros/README.md index df4d5b40ef93..d43ea4e6dccc 100644 --- a/lib/node_modules/@stdlib/array/zeros/README.md +++ b/lib/node_modules/@stdlib/array/zeros/README.md @@ -53,6 +53,7 @@ The function recognizes the following data types: - `float64`: double-precision floating-point numbers (IEEE 754) - `float32`: single-precision floating-point numbers (IEEE 754) +- `float16`: half-precision floating-point numbers (IEEE 754) - `complex128`: double-precision complex floating-point numbers - `complex64`: single-precision complex floating-point numbers - `int32`: 32-bit two's complement signed integers diff --git a/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.js b/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.js index e595991c28f0..60fb45b79895 100644 --- a/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.js +++ b/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.js @@ -84,6 +84,24 @@ bench( format( '%s:dtype=%s', pkg, 'float32' ), function benchmark( b ) { b.end(); }); +bench( format( '%s:dtype=%s', pkg, 'float16' ), function benchmark( b ) { + var arr; + var i; + b.tic(); + for ( i = 0; i < b.iterations; i++ ) { + arr = zeros( 0, 'float16' ); + if ( arr.length !== 0 ) { + b.fail( 'should have length 0' ); + } + } + b.toc(); + if ( !isTypedArrayLike( arr ) ) { + b.fail( 'should return a typed array' ); + } + b.pass( 'benchmark finished' ); + b.end(); +}); + bench( format( '%s:dtype=%s', pkg, 'complex128' ), function benchmark( b ) { var arr; var i; diff --git a/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.length.float16.js b/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.length.float16.js new file mode 100644 index 000000000000..5a2a72f01a1e --- /dev/null +++ b/lib/node_modules/@stdlib/array/zeros/benchmark/benchmark.length.float16.js @@ -0,0 +1,94 @@ +/** +* @license Apache-2.0 +* +* Copyright (c) 2022 The Stdlib Authors. +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +'use strict'; + +// MODULES // + +var bench = require( '@stdlib/bench' ); +var pow = require( '@stdlib/math/base/special/pow' ); +var isTypedArray = require( '@stdlib/assert/is-typed-array' ); +var format = require( '@stdlib/string/format' ); +var pkg = require( './../package.json' ).name; +var zeros = require( './../lib' ); + + +// FUNCTIONS // + +/** +* Creates a benchmark function. +* +* @private +* @param {PositiveInteger} len - array length +* @returns {Function} benchmark function +*/ +function createBenchmark( len ) { + return benchmark; + + /** + * Benchmark function. + * + * @private + * @param {Benchmark} b - benchmark instance + */ + function benchmark( b ) { + var arr; + var i; + + b.tic(); + for ( i = 0; i < b.iterations; i++ ) { + arr = zeros( len, 'float16' ); + if ( arr.length !== len ) { + b.fail( 'unexpected length' ); + } + } + b.toc(); + if ( !isTypedArray( arr ) ) { + b.fail( 'should return a typed array' ); + } + b.pass( 'benchmark finished' ); + b.end(); + } +} + + +// MAIN // + +/** +* Main execution sequence. +* +* @private +*/ +function main() { + var len; + var min; + var max; + var f; + var i; + + min = 1; // 10^min + max = 6; // 10^max + + for ( i = min; i <= max; i++ ) { + len = pow( 10, i ); + f = createBenchmark( len ); + bench( format( '%s:dtype=float16,len=%d', pkg, len ), f ); + } +} + +main(); diff --git a/lib/node_modules/@stdlib/array/zeros/docs/repl.txt b/lib/node_modules/@stdlib/array/zeros/docs/repl.txt index 6a593202a437..e368dca3314b 100644 --- a/lib/node_modules/@stdlib/array/zeros/docs/repl.txt +++ b/lib/node_modules/@stdlib/array/zeros/docs/repl.txt @@ -6,6 +6,7 @@ - float64: double-precision floating-point numbers (IEEE 754). - float32: single-precision floating-point numbers (IEEE 754). + - float16: half-precision floating-point numbers (IEEE 754). - complex128: double-precision complex floating-point numbers. - complex64: single-precision complex floating-point numbers. - int32: 32-bit two's complement signed integers. diff --git a/lib/node_modules/@stdlib/array/zeros/docs/types/index.d.ts b/lib/node_modules/@stdlib/array/zeros/docs/types/index.d.ts index 848339a517b0..12b3cd565897 100644 --- a/lib/node_modules/@stdlib/array/zeros/docs/types/index.d.ts +++ b/lib/node_modules/@stdlib/array/zeros/docs/types/index.d.ts @@ -29,6 +29,7 @@ import { NumericAndGenericDataTypeMap } from '@stdlib/types/array'; * * - `float64`: double-precision floating-point numbers (IEEE 754) * - `float32`: single-precision floating-point numbers (IEEE 754) +* - `float16`: half-precision floating-point numbers (IEEE 754) * - `complex128`: double-precision complex floating-point numbers * - `complex64`: single-precision complex floating-point numbers * - `int32`: 32-bit two's complement signed integers diff --git a/lib/node_modules/@stdlib/array/zeros/docs/types/test.ts b/lib/node_modules/@stdlib/array/zeros/docs/types/test.ts index a41620a03e07..a46ab1121fe7 100644 --- a/lib/node_modules/@stdlib/array/zeros/docs/types/test.ts +++ b/lib/node_modules/@stdlib/array/zeros/docs/types/test.ts @@ -26,6 +26,7 @@ import zeros = require( './index' ); zeros( 10 ); // $ExpectType Float64Array zeros( 10, 'float64' ); // $ExpectType Float64Array zeros( 10, 'float32' ); // $ExpectType Float32Array + zeros( 10, 'float16' ); // $ExpectType Float16Array zeros( 10, 'complex128' ); // $ExpectType Complex128Array zeros( 10, 'complex64' ); // $ExpectType Complex64Array zeros( 10, 'int32' ); // $ExpectType Int32Array diff --git a/lib/node_modules/@stdlib/array/zeros/package.json b/lib/node_modules/@stdlib/array/zeros/package.json index 4f8e94fb1444..00c1e2f1ed10 100644 --- a/lib/node_modules/@stdlib/array/zeros/package.json +++ b/lib/node_modules/@stdlib/array/zeros/package.json @@ -63,6 +63,7 @@ "matrix", "float64array", "float32array", + "float16array", "int32array", "uint32array", "int16array", @@ -84,6 +85,9 @@ "float", "single-precision", "float32", + "half", + "half-precision", + "float16", "ieee754", "integer", "int32", diff --git a/lib/node_modules/@stdlib/array/zeros/test/test.js b/lib/node_modules/@stdlib/array/zeros/test/test.js index 130ab209d7e1..f1fca430af80 100644 --- a/lib/node_modules/@stdlib/array/zeros/test/test.js +++ b/lib/node_modules/@stdlib/array/zeros/test/test.js @@ -23,6 +23,7 @@ var tape = require( 'tape' ); var Float64Array = require( '@stdlib/array/float64' ); var Float32Array = require( '@stdlib/array/float32' ); +var Float16Array = require( '@stdlib/array/float16' ); var Int32Array = require( '@stdlib/array/int32' ); var Uint32Array = require( '@stdlib/array/uint32' ); var Int16Array = require( '@stdlib/array/int16' ); @@ -183,6 +184,20 @@ tape( 'the function returns a zero-filled array (dtype=float32)', function test( t.end(); }); +tape( 'the function returns a zero-filled array (dtype=float16)', function test( t ) { + var expected; + var arr; + + expected = new Float16Array( [ 0.0, 0.0, 0.0, 0.0, 0.0 ] ); + + arr = zeros( 5, 'float16' ); + t.strictEqual( instanceOf( arr, Float16Array ), true, 'returns expected value' ); + t.strictEqual( arr.length, expected.length, 'returns expected value' ); + t.deepEqual( arr, expected, 'returns expected value' ); + + t.end(); +}); + tape( 'the function returns a zero-filled array (dtype=complex128)', function test( t ) { var expected; var arr;