Mbh 16.4.5

"'After his brother had thus departed from the (human) world, Vasudeva of celestial vision, who was fully acquainted with the end of all things, wandered for some time in that lonely forest thoughtfully. Endued with great energy he then sat down on the bare earth. He had thought before this of everything that had been fore-shadowed by the words uttered by Gandhari in former days. He also recollected the words that Durvasas had spoken at the time his body was smeared by that Rishi with the remnant of the Payasa he had eaten (while a guest at Krishna's house). The high-souled one, thinking of the destruction of the Vrishnis and the Andhakas, as also of the previous slaughter of the Kurus, concluded that the hour (for his own departure from the world) had come. He then restrained his senses (in Yoga). Conversant with the truth of every topic, Vasudeva, though he was the Supreme Deity, wished to die, for dispelling all doubts and establishing a certainty of results (in the matter of human existence), simply for upholding the three worlds and for making the words of Atri's son true. Having restrained all his senses, speech, and mind, Krishna laid himself down in high Yoga.