India remains in firm control of the second Test, with Sri Lanka struggling at 265/8 at stumps on Day 3, trailing by 238 runs.
Rishabh Pant retired hurt on Day 1 after being struck on the left arm by Lahiru Kumara.
Despite injury he starred with a brilliant 63 off 89 balls