[FEA]: C++ impl of Triton stage should cast input tensors to row-major if needed

[FEA]: C++ impl of Triton stage should cast input tensors to row-major if needed #406