As far as I know, what happens to Damon at the end happens in the third Infernal Affairs film.

I need to watch the original again, because I was tired and nearly dozed off. Mostly I remembered being dazzled by the visual design of the movie, such as the parking garage where the crime boss meets his end.