Skip to content

Commit c5f76be

Browse files
author
Rohan Yadav
committed
taco: add remaining frostt tensors to benchmark
1 parent 274961e commit c5f76be

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

taco/ufuncs.cpp

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,11 @@ static void bench_frostt_ufunc(benchmark::State& state, std::string tnsPath, Fun
206206
Tensor<int64_t> result("result", frosttTensor.getDimensions(), frosttTensor.getFormat());
207207
result.setAssembleWhileCompute(true);
208208
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+
}
209214
case 4: {
210215
IndexVar i, j, k, l;
211216
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());
242247
__func__(nips, "nips.tns") \
243248
__func__(uber_pickups, "uber-pickups.tns") \
244249
__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+
246263

247264
#define DECLARE_FROSTT_UFUNC_BENCH(name, path) \
248265
TACO_BENCH_ARGS(bench_frostt_ufunc, name/xor, path, xorOp); \

0 commit comments

Comments
 (0)