Skip to content

Commit 5e2c2d0

Browse files
committed
Remove dependency on change in core
1 parent a0500e5 commit 5e2c2d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/csrc/cuda/tensor_core_tiled_layout/libtorch.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ class VoidStarConverter: public c10::OperatorKernel {
5252
if (arg.isInt()) {
5353
ministack[idx] = reinterpret_cast<void *>(arg.toInt());
5454
} else if (arg.isTensor()) {
55-
const at::Tensor& tensor = arg.toTensor();
55+
at::Tensor& tensor = const_cast<at::Tensor&>(arg.toTensor());
5656
AtenTensorHandle ath = torch::aot_inductor::tensor_pointer_to_tensor_handle(&tensor);
5757
ministack[idx] = reinterpret_cast<void *>(ath);
5858
} else {

0 commit comments

Comments
 (0)