[Feature] Compatibility to DLPack 0.6 in tensoradapter (#3803)

* compatibility to DLPack 0.6 in tensoradapter

* fix

* oops
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>

tensoradapter/include/tensoradapter.h

@@ -13,12 +13,6 @@
 #include <dlpack/dlpack.h>
 #include <vector>
 
-#if defined(WIN32) || defined(_WIN32)
-#define TA_EXPORTS __declspec(dllexport)
-#else
-#define TA_EXPORTS
-#endif
-
 namespace tensoradapter {
 
 extern "C" {
@@ -31,7 +25,7 @@ extern "C" {
  * \param ctx The device
  * \return The allocated tensor
  */
-TA_EXPORTS DLManagedTensor* TAempty(
+DLManagedTensor* TAempty(
     std::vector<int64_t> shape, DLDataType dtype, DLContext ctx);
 
 }

tensoradapter/include/tensoradapter_exports.h

+/*!
+ *  Copyright (c) 2020 by Contributors
+ * \file tensoradapter_exports.h
+ * \brief Header file for functions exposed by the adapter library.
+ */
+
+#ifndef TENSORADAPTER_EXPORTS_H_
+#define TENSORADAPTER_EXPORTS_H_
+
+#if defined(WIN32) || defined(_WIN32)
+#define TA_EXPORTS __declspec(dllexport)
+#else
+#define TA_EXPORTS
+#endif
+
+#endif  // TENSORADAPTER_EXPORTS_H_

tensoradapter/pytorch/torch.cpp

@@ -4,12 +4,18 @@
  * \brief Implementation of PyTorch adapter library.
  */
 
-#include <tensoradapter.h>
+#include <tensoradapter_exports.h>
 #include <torch/torch.h>
 #include <ATen/DLConvertor.h>
 #include <vector>
 #include <iostream>
 
+#if DLPACK_VERSION > 040
+// Compatibility across DLPack - note that this assumes that the ABI stays the same.
+#define kDLGPU kDLCUDA
+#define DLContext DLDevice
+#endif
+
 namespace tensoradapter {
 
 static at::Device get_device(DLContext ctx) {
@@ -29,7 +35,7 @@ static at::Device get_device(DLContext ctx) {
 
 extern "C" {
 
-DLManagedTensor* TAempty(
+TA_EXPORTS DLManagedTensor* TAempty(
     std::vector<int64_t> shape,
     DLDataType dtype,
     DLContext ctx) {