fixes to SIMD and new benchmark testing float32x4.

hartsantler · hartsantler · commit d95187b82aad · 2014-06-13T00:08:50.000-07:00
diff --git a/pythonjs/python_to_pythonjs.py b/pythonjs/python_to_pythonjs.py
@@ -676,7 +676,7 @@ def visit_AugAssign(self, node):
 				raise NotImplementedError
 
 			b = '%s %s %s' %(target, op, self.visit(node.value))
-			if isinstance( node.target, ast.Name ) and node.target.id in self._typedef_vars and self._typedef_vars[node.target.id] in typedpython.native_number_types:
+			if isinstance( node.target, ast.Name ) and node.target.id in self._typedef_vars and self._typedef_vars[node.target.id] in typedpython.native_number_types+typedpython.vector_types:
 				writer.write(b)
 
 			else:
diff --git a/pythonjs/pythonjs.py b/pythonjs/pythonjs.py
@@ -349,6 +349,7 @@ def inline_helper_return_id(self, return_id):
 		return "var __returns__%s = null;"%return_id
 
 	def _visit_call_helper_numpy_array(self, node):
+		raise NotImplementedError('TODO numpy.array')
 		return self.visit(node.args[0])  ## TODO typed arrays
 
 	def _visit_call_helper_list(self, node):
diff --git a/regtests/bench/simd_float32x4.py b/regtests/bench/simd_float32x4.py
@@ -0,0 +1,15 @@
+"""simd float32x4 micro benchmark"""
+from time import time
+
+def main():
+	start = time()
+	float32x4 a = numpy.array( [1.0001, 1.0002, 1.0003, 1.0004], dtype=numpy.float32 )
+	float32x4 b = numpy.array( [1.00009, 1.00008, 1.00007, 1.00006], dtype=numpy.float32 )
+	float32x4 c = numpy.array( [1.00005, 1.00004, 1.00003, 1.00002], dtype=numpy.float32 )
+
+	arr = []
+	for i in range(20000):
+		c *= a*b
+		arr.append( a*b*c )
+
+	print(time()-start)
diff --git a/regtests/run.py b/regtests/run.py
@@ -348,6 +348,11 @@ def __call__(self, name):
         return lambda f: f
 webworker = webworker()
 
+try:
+    import numpy
+except:
+    import numpypy as numpy
+
 """
 
 def patch_python(filename, dart=False, python='PYTHONJS', backend=None):
diff --git a/regtests/typed/float32x4.py b/regtests/typed/float32x4.py
@@ -1,5 +1,4 @@
 """simd float32x4"""
-import numpy
 
 def main():
 	float32x4 a = numpy.array( [1.1, 1.2, 1.3, 0.4], dtype=numpy.float32 )

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`"""simd float32x4"""`
`2`		`-import numpy`
`3`	`2`
`4`	`3`	`def main():`
`5`	`4`	`float32x4 a = numpy.array( [1.1, 1.2, 1.3, 0.4], dtype=numpy.float32 )`