Skip to content

Commit 2d0ea39

Browse files
aman-095kgrytestdlib-bot
authored
feat: add C ndarray implementation for blas/base/csrot
PR-URL: #3067 Ref: #2039 Co-authored-by: Athan Reines <[email protected]> Reviewed-by: Athan Reines <[email protected]> Signed-off-by: Athan Reines <[email protected]> Co-authored-by: stdlib-bot <[email protected]>
1 parent 7462db7 commit 2d0ea39

File tree

11 files changed

+274
-76
lines changed

11 files changed

+274
-76
lines changed

‎lib/node_modules/@stdlib/blas/base/csrot/README.md

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -312,6 +312,33 @@ The function accepts the following arguments:
312312
void c_csrot( const CBLAS_INT N, void *CX, const CBLAS_INT strideX, void *CY, const CBLAS_INT strideY, const float c, const float s );
313313
```
314314

315+
#### c_csrot_ndarray( N, \*X, strideX, offsetX, \*Y, strideY, offsetY, c, s )
316+
317+
Applies a plane rotation using alternative indexing semantics.
318+
319+
```c
320+
float x[] = { 1.0f, 2.0f, 3.0f, 4.0f }; // interleaved real and imaginary components
321+
float y[] = { 5.0f, 6.0f, 7.0f, 8.0f };
322+
323+
c_csrot_ndarray( 2, (void *)x, 1, 0, (void *)y, 1, 0, 0.8f, 0.6f );
324+
```
325+
326+
The function accepts the following arguments:
327+
328+
- **N**: `[in] CBLAS_INT` number of indexed elements.
329+
- **CX**: `[inout] void*` first input array.
330+
- **strideX**: `[in] CBLAS_INT` index increment for `CX`.
331+
- **offsetX**: `[in] CBLAS_INT` starting index for `CX`.
332+
- **CY**: `[inout] void*` second input array.
333+
- **strideY**: `[in] CBLAS_INT` index increment for `CY`.
334+
- **offsetY**: `[in] CBLAS_INT` starting index for `CY`.
335+
- **c**: `[in] float` cosine of the angle of rotation.
336+
- **s**: `[in] float` sine of the angle of rotation.
337+
338+
```c
339+
void c_csrot_ndarray( const CBLAS_INT N, void *CX, const CBLAS_INT strideX, const CBLAS_INT offsetX, void *CY, const CBLAS_INT strideY, const CBLAS_INT offsetY, const float c, const float s );
340+
```
341+
315342
</section>
316343

317344
<!-- /.usage -->

‎lib/node_modules/@stdlib/blas/base/csrot/benchmark/c/benchmark.length.c

Lines changed: 44 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ static float rand_float( void ) {
9494
* @param len array length
9595
* @return elapsed time in seconds
9696
*/
97-
static double benchmark( int iterations, int len ) {
97+
static double benchmark1( int iterations, int len ) {
9898
double elapsed;
9999
float x[ len*2 ];
100100
float y[ len*2 ];
@@ -122,6 +122,41 @@ static double benchmark( int iterations, int len ) {
122122
return elapsed;
123123
}
124124

125+
/**
126+
* Runs a benchmark.
127+
*
128+
* @param iterations number of iterations
129+
* @param len array length
130+
* @return elapsed time in seconds
131+
*/
132+
static double benchmark2( int iterations, int len ) {
133+
double elapsed;
134+
float x[ len*2 ];
135+
float y[ len*2 ];
136+
double t;
137+
int i;
138+
139+
for ( i = 0; i < len; i++ ) {
140+
x[ i ] = ( rand_float()*10000.0f ) - 5000.0f;
141+
x[ i+1 ] = ( rand_float()*10000.0f ) - 5000.0f;
142+
y[ i ] = ( rand_float()*10000.0f ) - 5000.0f;
143+
y[ i+1 ] = ( rand_float()*10000.0f ) - 5000.0f;
144+
}
145+
t = tic();
146+
for ( i = 0; i < iterations; i++ ) {
147+
c_csrot_ndarray( len, (void *)x, 1, 0, (void *)y, 1, 0, 0.8f, 0.6f );
148+
if ( y[ 0 ] != y[ 0 ] ) {
149+
printf( "should not return NaN\n" );
150+
break;
151+
}
152+
}
153+
elapsed = tic() - t;
154+
if ( y[ 0 ] != y[ 0 ] ) {
155+
printf( "should not return NaN\n" );
156+
}
157+
return elapsed;
158+
}
159+
125160
/**
126161
* Main execution sequence.
127162
*/
@@ -144,7 +179,14 @@ int main( void ) {
144179
for ( j = 0; j < REPEATS; j++ ) {
145180
count += 1;
146181
printf( "# c::%s:len=%d\n", NAME, len );
147-
elapsed = benchmark( iter, len );
182+
elapsed = benchmark1( iter, len );
183+
print_results( iter, elapsed );
184+
printf( "ok %d benchmark finished\n", count );
185+
}
186+
for ( j = 0; j < REPEATS; j++ ) {
187+
count += 1;
188+
printf( "# c::%s:ndarray:len=%d\n", NAME, len );
189+
elapsed = benchmark2( iter, len );
148190
print_results( iter, elapsed );
149191
printf( "ok %d benchmark finished\n", count );
150192
}

‎lib/node_modules/@stdlib/blas/base/csrot/examples/c/example.c

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,12 @@ int main( void ) {
3838
printf( "x[ %i ] = %f + %fj\n", i, x[ i*2 ], x[ (i*2)+1 ] );
3939
printf( "y[ %i ] = %f + %fj\n", i, y[ i*2 ], y[ (i*2)+1 ] );
4040
}
41+
42+
c_csrot_ndarray( N, (void *)x, -strideX, N-1, (void *)y, strideY, N-1, 0.8f, 0.6f );
43+
44+
// Print the result:
45+
for ( int i = 0; i < N; i++ ) {
46+
printf( "x[ %i ] = %f + %fj\n", i, x[ i*2 ], x[ (i*2)+1 ] );
47+
printf( "y[ %i ] = %f + %fj\n", i, y[ i*2 ], y[ (i*2)+1 ] );
48+
}
4149
}

‎lib/node_modules/@stdlib/blas/base/csrot/include/stdlib/blas/base/csrot.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,11 @@ extern "C" {
3636
*/
3737
void API_SUFFIX(c_csrot)( const CBLAS_INT N, void *CX, const CBLAS_INT strideX, void *CY, const CBLAS_INT strideY, const float c, const float s );
3838

39+
/**
40+
* Applies a plane rotation using alternative indexing semantics.
41+
*/
42+
void API_SUFFIX(c_csrot_ndarray)( const CBLAS_INT N, void *CX, const CBLAS_INT strideX, const CBLAS_INT offsetX, void *CY, const CBLAS_INT strideY, const CBLAS_INT offsetY, const float c, const float s );
43+
3944
#ifdef __cplusplus
4045
}
4146
#endif

‎lib/node_modules/@stdlib/blas/base/csrot/lib/ndarray.native.js

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
// MODULES //
2222

2323
var reinterpret = require( '@stdlib/strided/base/reinterpret-complex64' );
24-
var minViewBufferIndex = require( '@stdlib/strided/base/min-view-buffer-index' );
2524
var addon = require( './../src/addon.node' );
2625

2726

@@ -70,16 +69,9 @@ var addon = require( './../src/addon.node' );
7069
* // returns ~1.6
7170
*/
7271
function csrot( N, cx, strideX, offsetX, cy, strideY, offsetY, c, s ) {
73-
var viewX;
74-
var viewY;
75-
76-
offsetX = minViewBufferIndex( N, strideX, offsetX );
77-
offsetY = minViewBufferIndex( N, strideY, offsetY );
78-
79-
viewX = reinterpret( cx, offsetX );
80-
viewY = reinterpret( cy, offsetY );
81-
82-
addon( N, viewX, strideX, viewY, strideY, c, s );
72+
var viewX = reinterpret( cx, 0 );
73+
var viewY = reinterpret( cy, 0 );
74+
addon.ndarray( N, viewX, strideX, offsetX, viewY, strideY, offsetY, c, s );
8375
return cy;
8476
}
8577

‎lib/node_modules/@stdlib/blas/base/csrot/manifest.json

Lines changed: 48 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -45,10 +45,12 @@
4545
"dependencies": [
4646
"@stdlib/napi/export",
4747
"@stdlib/napi/argv",
48+
"@stdlib/strided/base/min-view-buffer-index",
4849
"@stdlib/napi/argv-int64",
4950
"@stdlib/napi/argv-float",
5051
"@stdlib/napi/argv-strided-complex64array",
51-
"@stdlib/blas/base/shared"
52+
"@stdlib/blas/base/shared",
53+
"@stdlib/complex/float32/ctor"
5254
]
5355
},
5456
{
@@ -57,7 +59,8 @@
5759
"blas": "",
5860
"wasm": false,
5961
"src": [
60-
"./src/csrot.c"
62+
"./src/csrot.c",
63+
"./src/csrot_ndarray.c"
6164
],
6265
"include": [
6366
"./include"
@@ -75,7 +78,8 @@
7578
"blas": "",
7679
"wasm": false,
7780
"src": [
78-
"./src/csrot.c"
81+
"./src/csrot.c",
82+
"./src/csrot_ndarray.c"
7983
],
8084
"include": [
8185
"./include"
@@ -107,10 +111,12 @@
107111
"dependencies": [
108112
"@stdlib/napi/export",
109113
"@stdlib/napi/argv",
114+
"@stdlib/strided/base/min-view-buffer-index",
110115
"@stdlib/napi/argv-int64",
111116
"@stdlib/napi/argv-float",
112117
"@stdlib/napi/argv-strided-complex64array",
113-
"@stdlib/blas/base/shared"
118+
"@stdlib/blas/base/shared",
119+
"@stdlib/complex/float32/ctor"
114120
]
115121
},
116122
{
@@ -130,7 +136,9 @@
130136
],
131137
"libpath": [],
132138
"dependencies": [
133-
"@stdlib/blas/base/shared"
139+
"@stdlib/blas/base/shared",
140+
"@stdlib/strided/base/min-view-buffer-index",
141+
"@stdlib/complex/float32/ctor"
134142
]
135143
},
136144
{
@@ -150,7 +158,9 @@
150158
],
151159
"libpath": [],
152160
"dependencies": [
153-
"@stdlib/blas/base/shared"
161+
"@stdlib/blas/base/shared",
162+
"@stdlib/strided/base/min-view-buffer-index",
163+
"@stdlib/complex/float32/ctor"
154164
]
155165
},
156166

@@ -171,10 +181,12 @@
171181
"dependencies": [
172182
"@stdlib/napi/export",
173183
"@stdlib/napi/argv",
184+
"@stdlib/strided/base/min-view-buffer-index",
174185
"@stdlib/napi/argv-int64",
175186
"@stdlib/napi/argv-float",
176187
"@stdlib/napi/argv-strided-complex64array",
177-
"@stdlib/blas/base/shared"
188+
"@stdlib/blas/base/shared",
189+
"@stdlib/complex/float32/ctor"
178190
]
179191
},
180192
{
@@ -183,7 +195,8 @@
183195
"blas": "",
184196
"wasm": false,
185197
"src": [
186-
"./src/csrot.c"
198+
"./src/csrot.c",
199+
"./src/csrot_ndarray.c"
187200
],
188201
"include": [
189202
"./include"
@@ -201,7 +214,8 @@
201214
"blas": "",
202215
"wasm": false,
203216
"src": [
204-
"./src/csrot.c"
217+
"./src/csrot.c",
218+
"./src/csrot_ndarray.c"
205219
],
206220
"include": [
207221
"./include"
@@ -232,10 +246,12 @@
232246
"dependencies": [
233247
"@stdlib/napi/export",
234248
"@stdlib/napi/argv",
249+
"@stdlib/strided/base/min-view-buffer-index",
235250
"@stdlib/napi/argv-int64",
236251
"@stdlib/napi/argv-float",
237252
"@stdlib/napi/argv-strided-complex64array",
238-
"@stdlib/blas/base/shared"
253+
"@stdlib/blas/base/shared",
254+
"@stdlib/complex/float32/ctor"
239255
]
240256
},
241257
{
@@ -254,7 +270,9 @@
254270
],
255271
"libpath": [],
256272
"dependencies": [
257-
"@stdlib/blas/base/shared"
273+
"@stdlib/blas/base/shared",
274+
"@stdlib/strided/base/min-view-buffer-index",
275+
"@stdlib/complex/float32/ctor"
258276
]
259277
},
260278
{
@@ -273,7 +291,9 @@
273291
],
274292
"libpath": [],
275293
"dependencies": [
276-
"@stdlib/blas/base/shared"
294+
"@stdlib/blas/base/shared",
295+
"@stdlib/strided/base/min-view-buffer-index",
296+
"@stdlib/complex/float32/ctor"
277297
]
278298
},
279299

@@ -299,7 +319,8 @@
299319
"@stdlib/napi/argv-int64",
300320
"@stdlib/napi/argv-float",
301321
"@stdlib/napi/argv-strided-complex64array",
302-
"@stdlib/blas/base/shared"
322+
"@stdlib/blas/base/shared",
323+
"@stdlib/complex/float32/ctor"
303324
]
304325
},
305326
{
@@ -319,7 +340,9 @@
319340
],
320341
"libpath": [],
321342
"dependencies": [
322-
"@stdlib/blas/base/shared"
343+
"@stdlib/blas/base/shared",
344+
"@stdlib/strided/base/min-view-buffer-index",
345+
"@stdlib/complex/float32/ctor"
323346
]
324347
},
325348
{
@@ -339,7 +362,9 @@
339362
],
340363
"libpath": [],
341364
"dependencies": [
342-
"@stdlib/blas/base/shared"
365+
"@stdlib/blas/base/shared",
366+
"@stdlib/strided/base/min-view-buffer-index",
367+
"@stdlib/complex/float32/ctor"
343368
]
344369
},
345370

@@ -349,7 +374,8 @@
349374
"blas": "",
350375
"wasm": false,
351376
"src": [
352-
"./src/csrot.c"
377+
"./src/csrot.c",
378+
"./src/csrot_ndarray.c"
353379
],
354380
"include": [
355381
"./include"
@@ -372,7 +398,8 @@
372398
"blas": "",
373399
"wasm": false,
374400
"src": [
375-
"./src/csrot.c"
401+
"./src/csrot.c",
402+
"./src/csrot_ndarray.c"
376403
],
377404
"include": [
378405
"./include"
@@ -390,7 +417,8 @@
390417
"blas": "",
391418
"wasm": false,
392419
"src": [
393-
"./src/csrot.c"
420+
"./src/csrot.c",
421+
"./src/csrot_ndarray.c"
394422
],
395423
"include": [
396424
"./include"
@@ -409,7 +437,8 @@
409437
"blas": "",
410438
"wasm": true,
411439
"src": [
412-
"./src/csrot.c"
440+
"./src/csrot.c",
441+
"./src/csrot_ndarray.c"
413442
],
414443
"include": [
415444
"./include"

‎lib/node_modules/@stdlib/blas/base/csrot/src/addon.c

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,4 +45,26 @@ static napi_value addon( napi_env env, napi_callback_info info ) {
4545
return NULL;
4646
}
4747

48-
STDLIB_NAPI_MODULE_EXPORT_FCN( addon )
48+
/**
49+
* Receives JavaScript callback invocation data.
50+
*
51+
* @param env environment under which the function is invoked
52+
* @param info callback data
53+
* @return Node-API value
54+
*/
55+
static napi_value addon_method( napi_env env, napi_callback_info info ) {
56+
STDLIB_NAPI_ARGV( env, info, argv, argc, 9 );
57+
STDLIB_NAPI_ARGV_INT64( env, N, argv, 0 );
58+
STDLIB_NAPI_ARGV_INT64( env, strideX, argv, 2 );
59+
STDLIB_NAPI_ARGV_INT64( env, offsetX, argv, 3 );
60+
STDLIB_NAPI_ARGV_INT64( env, strideY, argv, 5 );
61+
STDLIB_NAPI_ARGV_INT64( env, offsetY, argv, 6 );
62+
STDLIB_NAPI_ARGV_FLOAT( env, c, argv, 7 );
63+
STDLIB_NAPI_ARGV_FLOAT( env, s, argv, 8 );
64+
STDLIB_NAPI_ARGV_STRIDED_COMPLEX64ARRAY( env, CX, N, strideX, argv, 1 );
65+
STDLIB_NAPI_ARGV_STRIDED_COMPLEX64ARRAY( env, CY, N, strideY, argv, 4 );
66+
API_SUFFIX(c_csrot_ndarray)( N, (void *)CX, strideX, offsetX, (void *)CY, strideY, offsetY, c, s );
67+
return NULL;
68+
}
69+
70+
STDLIB_NAPI_MODULE_EXPORT_FCN_WITH_METHOD( addon, "ndarray", addon_method )

0 commit comments

Comments
 (0)