Skip to content

Commit 0828553

Browse files
authored
[TVM] Rename NDArray -> Tensor (#1651)
## 📌 Description The latest TVM ffi refactor renames `NDArray` to `Tensor`, to align with the terminology in PyTorch. This PR updates flashinfer tvm binding with the rename. apache/tvm#18275 ## 🔍 Related Issues N/A ## 🚀 Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### ✅ Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## 🧪 Tests N/A ## Reviewer Notes N/A
1 parent a42f992 commit 0828553

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tvm_binding/tvm_binding_utils.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@
2121
#include <tvm/ffi/function.h>
2222
#include <tvm/runtime/data_type.h>
2323
#include <tvm/runtime/int_tuple.h>
24-
#include <tvm/runtime/ndarray.h>
24+
#include <tvm/runtime/tensor.h>
2525

2626
using IdType = int32_t;
2727
using tvm::ffi::Array;
2828
using tvm::runtime::DataType;
2929
using tvm::runtime::IntTuple;
30-
using tvm::runtime::NDArray;
30+
using tvm::runtime::Tensor;
3131

3232
#define DISPATCH_BOOL(expr, const_expr, ...) \
3333
[&]() -> bool { \

0 commit comments

Comments
 (0)