@@ -128,7 +128,7 @@ def load(self, tensor, suiteSparse):
128
128
# Run benchmarks against the FROSTT collection.
129
129
FROSTTTensors = TensorCollectionFROSTT ()
130
130
@pytest .mark .parametrize ("tensor" , FROSTTTensors .getTensors ())
131
- @pytest .mark .parametrize ("ufunc" , [numpy .bitwise_xor , numpy .ldexp , numpy .right_shift ])
131
+ @pytest .mark .parametrize ("ufunc" , [numpy .logical_xor , numpy .ldexp , numpy .right_shift ])
132
132
def bench_pydata_frostt_ufunc_sparse (tacoBench , tensor , ufunc ):
133
133
frTensor , other = inputCache .load (tensor , False )
134
134
def bench ():
@@ -145,7 +145,7 @@ def bench():
145
145
else :
146
146
tacoBench (bench , extra_info )
147
147
148
- fusedFuncs = [lambda a , b , c : numpy .bitwise_and (numpy .bitwise_xor (a , b ), c )]
148
+ fusedFuncs = [lambda a , b , c : numpy .logical_and (numpy .logical_xor (a , b ), c )]
149
149
fusedFuncNames = ["xorAndFused" ]
150
150
fusedFuncs = zip (fusedFuncs , fusedFuncNames )
151
151
@pytest .mark .parametrize ("tensor" , FROSTTTensors .getTensors ())
@@ -162,7 +162,7 @@ def bench():
162
162
tacoBench (bench , extra_info )
163
163
164
164
# Run benchmarks against the SuiteSparse collection.
165
- @pytest .mark .parametrize ("ufunc" , [numpy .bitwise_xor , numpy .ldexp , numpy .right_shift ])
165
+ @pytest .mark .parametrize ("ufunc" , [numpy .logical_xor , numpy .ldexp , numpy .right_shift ])
166
166
def bench_pydata_suitesparse_ufunc_sparse (tacoBench , ufunc ):
167
167
tensor = SuiteSparseTensor (os .getenv ('SUITESPARSE_TENSOR_PATH' ))
168
168
ssTensor , other = inputCache .load (tensor , True )
0 commit comments