Added SuiteSparse GraphBLAS as optional dependency

Author: stephenchouca

.gitmodules

@@ -4,3 +4,6 @@
 [submodule "taco/benchmark"]
 	path = taco/benchmark
 	url = ../../google/benchmark.git
+[submodule "taco/suitesparse"]
+	path = taco/suitesparse
+	url = ../../DrTimothyAldenDavis/GraphBLAS.git

Makefile

@@ -5,6 +5,9 @@ BENCHFLAGS := #"--benchmark-group-by=func"
 IGNORE += taco
 IGNORE_FLAGS := $(addprefix --ignore=, $(IGNORE)) 
 
+GRAPHBLAS := "OFF"
+OPENMP := "OFF"
+
 export TACO_TENSOR_PATH = data/
 
 # To group benchmark output by benchmark, use BENCHFLAGS=--benchmark-group-by=func.
@@ -21,7 +24,7 @@ else
 endif
 
 taco/build/taco-bench: check-and-reinit-submodules taco/benchmark/googletest
-	mkdir -p taco/build/ && cd taco/build/ && cmake ../ && make -j4 taco-bench 
+	mkdir -p taco/build/ && cd taco/build/ && cmake -DOPENMP=$(OPENMP) -DGRAPHBLAS=$(GRAPHBLAS) ../ && $(MAKE) taco-bench 
 
 taco/benchmark/googletest: check-and-reinit-submodules
 	if [ ! -d "taco/benchmark/googletest" ] ; then git clone https://github.com/google/googletest taco/benchmark/googletest; fi

taco/CMakeLists.txt

@@ -1,11 +1,20 @@
 cmake_minimum_required(VERSION 2.8.12 FATAL_ERROR)
 project(taco-bench)
 set(CMAKE_BUILD_TYPE "release")
+option(GRAPHBLAS "Build GraphBLAS benchmarks" OFF)
+
+if(GRAPHBLAS)
+  message("-- Will build GraphBLAS benchmarks")
+  add_definitions(-DGRAPHBLAS)
+endif(GRAPHBLAS)
 
 add_subdirectory(taco)
 add_subdirectory(benchmark)
+if(GRAPHBLAS)
+  add_subdirectory(suitesparse)
+endif(GRAPHBLAS)
 
-include_directories(taco taco/include benchmark/include)
+include_directories(taco taco/include benchmark/include suitesparse/Include)
 
 file(GLOB TEST_SOURCES *.cpp *.h)
 
@@ -14,4 +23,7 @@ set(CMAKE_CXX_FLAGS "${C_CXX_FLAGS} -std=c++14")
 add_executable(taco-bench ${TEST_SOURCES} bench.h)
 target_link_libraries(taco-bench benchmark::benchmark)
 target_link_libraries(taco-bench taco)
+if(GRAPHBLAS)
+  target_link_libraries(taco-bench graphblas)
+endif(GRAPHBLAS)
 

taco/graphblas.cpp

@@ -1,3 +1,5 @@
+#if GRAPHBLAS
+
 #include "bench.h"
 #include "benchmark/benchmark.h"
 
@@ -6,6 +8,10 @@
 #include "taco/index_notation/index_notation.h"
 #include "taco/index_notation/tensor_operator.h"
 
+extern "C" {
+#include "GraphBLAS.h"
+}
+
 #include <vector>
 #include <limits>
 
@@ -76,4 +82,24 @@ static void bench_mxv_taco(benchmark::State& state) {
   taco_set_num_threads(1);
 }
 
+static void bench_mxv_suitesparse(benchmark::State& state) {
+  GrB_init(GrB_BLOCKING);
+
+  Tensor<double> T = read("/data/scratch/s3chou/formats-bench/data/webbase_1M.mtx", CSR);
+
+  for (const auto& c : T) {
+    //A.insert(c.first.toVector(), c.second);
+  }
+
+  GrB_Vector x = nullptr;
+  GrB_Index n;
+  GrB_Vector_new(&x, GrB_FP64, n);
+
+  for (auto _ : state) {
+  }
+}
+
 TACO_BENCH(bench_mxv_taco);
+TACO_BENCH(bench_mxv_suitesparse);
+
+#endif