Archimedes' grave was actually described in ancient texts.
Sadly, the movie missed a chance to recreate an authentic historical artifact. It was a short column topped by a cylinder and a sphere, as depicted at the lower left of this painting:
https://commons.m.wikimedia.org/wiki/File:Cicero_entdeckt_das_Grabmal_des_Archimedes_(Zuccarelli).jpg