Make cuTENSOR available in cupy.einsum
@@ -303,9 +303,6 @@ def _use_cutensor(dtype0, sub0, dtype1, sub1, batch_dims, contract_dims):
if dtype0 not in (cupy.float32, cupy.float64,
cupy.complex64, cupy.complex128):
return False
- if (len(contract_dims) >= 1 and (sub0[-1] in batch_dims or
- sub1[-1] in batch_dims)):
- return False
return True
