@@ -206,6 +206,11 @@ static void bench_frostt_ufunc(benchmark::State& state, std::string tnsPath, Fun
206
206
Tensor<int64_t > result (" result" , frosttTensor.getDimensions (), frosttTensor.getFormat ());
207
207
result.setAssembleWhileCompute (true );
208
208
switch (frosttTensor.getOrder ()) {
209
+ case 3 : {
210
+ IndexVar i, j, k;
211
+ result (i, j, k) = op (frosttTensor (i, j, k), other (i, j, k));
212
+ break ;
213
+ }
209
214
case 4 : {
210
215
IndexVar i, j, k, l;
211
216
result (i, j, k, l) = op (frosttTensor (i, j, k, l), other (i, j, k, l));
@@ -242,7 +247,19 @@ Func xorOp("logical_xor", GeneralAdd(), xorAlgebra());
242
247
__func__ (nips, " nips.tns" ) \
243
248
__func__(uber_pickups, " uber-pickups.tns" ) \
244
249
__func__(chicago_crime, " chicago-crime.tns" ) \
245
- __func__(lbnl_network, " lbnl-network.tns" )
250
+ __func__(lbnl_network, " lbnl-network.tns" ) \
251
+ __func__(enron, " enron.tns" ) \
252
+ __func__(nell-2 , " nell-2.tns" ) \
253
+ __func__(vast, " vast.tns" )
254
+
255
+ // Other FROSTT tensors that may or may not be too large to load.
256
+ // __func__(delicious, "delicious.tns") \
257
+ // __func__(flickr, "flickr.tns") \
258
+ // __func__(nell-1, "nell-1.tns") \
259
+ // __func__(patents, "patents.tns") \
260
+ // __func__(reddit, "reddit.tns") \
261
+ // __func__(amazon-reviews, "amazon-reviews.tns") \
262
+
246
263
247
264
#define DECLARE_FROSTT_UFUNC_BENCH (name, path ) \
248
265
TACO_BENCH_ARGS (bench_frostt_ufunc, name/xor , path, xorOp); \
0 commit comments