Skip to content

Commit 1d2ff3a

Browse files
committed
Remove an incorrect check
1 parent aa0b109 commit 1d2ff3a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tilelang/jit/adapter/cython/cython_wrapper.pyx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,12 +197,14 @@ cdef class CythonKernelWrapper:
197197
tensor = inputs[ins_idx]
198198
ins_idx += 1
199199
# TODO(chenggang): remove this check or rewrite by ourselves?
200+
'''
200201
if isinstance(tensor, torch.Tensor) and tensor._base is not None and not tensor.is_contiguous():
201202
base_tensor = tensor._base.as_strided(tensor._base.shape, tensor.stride())
202203
if torch._debug_has_internal_overlap(base_tensor):
203204
raise ValueError(f"Cannot use an overlapping tensor"
204205
f"(shape={tensor.shape}, strides={tensor.stride()}, "
205206
f"overlap={torch._debug_has_internal_overlap(base_tensor)}) as the kernel input")
207+
'''
206208
tensor_list.append(tensor)
207209

208210
# Convert tensor pointers to C void pointers for kernel call

0 commit comments

Comments
 (0)