Unverified Commit 6f6ef7ad authored by LJC00118's avatar LJC00118 Committed by GitHub
Browse files

[Cython] Remove an incorrect check (#880)

parent aa0b1090
...@@ -197,12 +197,14 @@ cdef class CythonKernelWrapper: ...@@ -197,12 +197,14 @@ cdef class CythonKernelWrapper:
tensor = inputs[ins_idx] tensor = inputs[ins_idx]
ins_idx += 1 ins_idx += 1
# TODO(chenggang): remove this check or rewrite by ourselves? # TODO(chenggang): remove this check or rewrite by ourselves?
'''
if isinstance(tensor, torch.Tensor) and tensor._base is not None and not tensor.is_contiguous(): if isinstance(tensor, torch.Tensor) and tensor._base is not None and not tensor.is_contiguous():
base_tensor = tensor._base.as_strided(tensor._base.shape, tensor.stride()) base_tensor = tensor._base.as_strided(tensor._base.shape, tensor.stride())
if torch._debug_has_internal_overlap(base_tensor): if torch._debug_has_internal_overlap(base_tensor):
raise ValueError(f"Cannot use an overlapping tensor" raise ValueError(f"Cannot use an overlapping tensor"
f"(shape={tensor.shape}, strides={tensor.stride()}, " f"(shape={tensor.shape}, strides={tensor.stride()}, "
f"overlap={torch._debug_has_internal_overlap(base_tensor)}) as the kernel input") f"overlap={torch._debug_has_internal_overlap(base_tensor)}) as the kernel input")
'''
tensor_list.append(tensor) tensor_list.append(tensor)
# Convert tensor pointers to C void pointers for kernel call # Convert tensor pointers to C void pointers for kernel call
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment