We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a0500e5 commit 5e2c2d0Copy full SHA for 5e2c2d0
torchao/csrc/cuda/tensor_core_tiled_layout/libtorch.cpp
@@ -52,7 +52,7 @@ class VoidStarConverter: public c10::OperatorKernel {
52
if (arg.isInt()) {
53
ministack[idx] = reinterpret_cast<void *>(arg.toInt());
54
} else if (arg.isTensor()) {
55
- const at::Tensor& tensor = arg.toTensor();
+ at::Tensor& tensor = const_cast<at::Tensor&>(arg.toTensor());
56
AtenTensorHandle ath = torch::aot_inductor::tensor_pointer_to_tensor_handle(&tensor);
57
ministack[idx] = reinterpret_cast<void *>(ath);
58
} else {
0 commit comments