find_package(TensorFlow)
if(${TensorFlow_FOUND})
  set(CUDA_HOME "/usr/local/cuda")
  include_directories("${TF_INC}/tensorflow/include")
  include_directories("${CUDA_HOME}/include")
  link_directories(${TF_LIB})
  add_definitions(-D_GLIBCXX_USE_CXX11_ABI=${TF_ABI})
  add_library(tf_blocksparse SHARED dot.cpp)
  target_link_libraries(tf_blocksparse tensorflow_framework triton)
endif()