FilesExpand file tree

main

/

test-wasm.mjs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

176 lines (137 loc) · 6.12 KB

main

/

test-wasm.mjs

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

176 lines (137 loc) · 6.12 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

#!/usr/bin/env node

/**

* WASM Package Test Script

* Tests ruvector-math-wasm and ruvector-attention-wasm in Node.js

*/

import { readFileSync } from 'fs';

import { dirname, join } from 'path';

import { fileURLToPath } from 'url';

const __dirname = dirname(fileURLToPath(import.meta.url));

console.log('🧪 Testing RuVector WASM Packages\n');

// ============================================================================

// Test ruvector-math-wasm

// ============================================================================

async function testMathWasm() {

console.log('📦 Testing ruvector-math-wasm...');

const pkgPath = join(__dirname, '../crates/ruvector-math-wasm/pkg');

try {

// Load WASM module

const wasmPath = join(pkgPath, 'ruvector_math_wasm_bg.wasm');

const wasmBuffer = readFileSync(wasmPath);

// Import the JS bindings

const mathWasm = await import(join(pkgPath, 'ruvector_math_wasm.js'));

// Initialize with WASM bytes

await mathWasm.default(wasmBuffer);

// Test 1: Sliced Wasserstein Distance

console.log(' ├─ Testing SlicedWasserstein...');

const sw = new mathWasm.WasmSlicedWasserstein(100);

// Create test point clouds (3 points in 2D each)

const source = new Float64Array([0, 0, 1, 0, 0, 1]);

const target = new Float64Array([2, 0, 3, 0, 2, 1]);

const distance = sw.distance(source, target, 2);

console.log(` │ Distance: ${distance.toFixed(4)}`);

if (distance > 0 && distance < 10) {

console.log(' │ ✅ SlicedWasserstein works!');

} else {

throw new Error(`Unexpected distance: ${distance}`);

}

// Test 2: Product Manifold

console.log(' ├─ Testing ProductManifold...');

const manifold = new mathWasm.WasmProductManifold(4, 2, 2); // E^4 x H^2 x S^2

// Create test points (8D total)

const pointA = new Float64Array([1, 0, 0, 0, 0.1, 0.1, 1, 0]);

const pointB = new Float64Array([0, 1, 0, 0, 0.2, 0.1, 0, 1]);

const manifoldDist = manifold.distance(pointA, pointB);

console.log(` │ Manifold distance: ${manifoldDist.toFixed(4)}`);

if (manifoldDist > 0) {

console.log(' │ ✅ ProductManifold works!');

} else {

throw new Error(`Unexpected manifold distance: ${manifoldDist}`);

}

// Test 3: Spherical Space

console.log(' ├─ Testing SphericalSpace...');

const sphere = new mathWasm.WasmSphericalSpace(3);

const vecA = new Float64Array([1, 0, 0]);

const vecB = new Float64Array([0, 1, 0]);

const sphereDist = sphere.distance(vecA, vecB);

console.log(` │ Spherical distance: ${sphereDist.toFixed(4)} (expected: ~1.5708 = π/2)`);

if (Math.abs(sphereDist - Math.PI/2) < 0.01) {

console.log(' │ ✅ SphericalSpace works!');

} else {

throw new Error(`Unexpected spherical distance: ${sphereDist}`);

}

console.log(' └─ ✅ ruvector-math-wasm: All tests passed!\n');

return true;

} catch (error) {

console.error(' └─ ❌ Error:', error.message);

return false;

}

}

// ============================================================================

// Test ruvector-attention-wasm

// ============================================================================

async function testAttentionWasm() {

console.log('📦 Testing ruvector-attention-wasm...');

const pkgPath = join(__dirname, '../crates/ruvector-attention-wasm/pkg');

try {

// Check if pkg exists (need to build first)

const wasmPath = join(pkgPath, 'ruvector_attention_wasm_bg.wasm');

let wasmBuffer;

try {

wasmBuffer = readFileSync(wasmPath);

} catch {

console.log(' └─ ⚠️ Package not built. Building now...');

const { execSync } = await import('child_process');

execSync('wasm-pack build crates/ruvector-attention-wasm --target web --release', {

cwd: join(__dirname, '..'),

stdio: 'inherit'

});

wasmBuffer = readFileSync(wasmPath);

}

// Import the JS bindings

const attentionWasm = await import(join(pkgPath, 'ruvector_attention_wasm.js'));

// Initialize with WASM bytes

await attentionWasm.default(wasmBuffer);

// Test 1: Scaled Dot Product Attention

console.log(' ├─ Testing ScaledDotProductAttention...');

if (attentionWasm.WasmScaledDotProductAttention) {

const attention = new attentionWasm.WasmScaledDotProductAttention(64);

console.log(' │ ✅ ScaledDotProductAttention initialized');

} else {

console.log(' │ ⚠️ ScaledDotProductAttention not exported');

}

// Test 2: Flash Attention (if available)

console.log(' ├─ Testing FlashAttention...');

if (attentionWasm.WasmFlashAttention) {

const flash = new attentionWasm.WasmFlashAttention(64, 64);

console.log(' │ ✅ FlashAttention initialized');

} else {

console.log(' │ ⚠️ FlashAttention not exported');

}

// List available exports

console.log(' ├─ Available exports:');

const exports = Object.keys(attentionWasm).filter(k => k.startsWith('Wasm'));

exports.forEach(e => console.log(` │ - ${e}`));

console.log(' └─ ✅ ruvector-attention-wasm: Package loaded successfully!\n');

return true;

} catch (error) {

console.error(' └─ ❌ Error:', error.message);

return false;

}

}

// ============================================================================

// Run all tests

// ============================================================================

async function main() {

const results = {

math: await testMathWasm(),

attention: await testAttentionWasm()

};

console.log('═══════════════════════════════════════');

console.log('📊 Test Results:');

console.log(` ruvector-math-wasm: ${results.math ? '✅ PASS' : '❌ FAIL'}`);

console.log(` ruvector-attention-wasm: ${results.attention ? '✅ PASS' : '❌ FAIL'}`);

console.log('═══════════════════════════════════════');

process.exit(results.math && results.attention ? 0 : 1);

}

main().catch(console.error);