Sri Lanka got off to a disastrous start as they collapsed to three for four after all the first three batsmen dismissed were out for nought, with both openers falling for golden ducks.
Bumrah bowled two wides to give Sri Lanka their first runs, but at the start of the next over Mohammed Siraj had Dimuth Karunaratne, the other opening batsman, lbw for nought as well.
Sadeera Samarawickrama, batting at number four, managed to survive for just four deliveries before he too fell for a duck, edging the ball to third slip, where Shreyas Iyer completed the catch.
Sri Lanka captain Kusal Mendis faced a similar fate when Siraj bowled him for just one run, leaving his team reeling at three for four in the fourth over.