Adding flex_attention benchmark for eager and compile mode (#174)

Adding flex_attention benchmark for eager and compile mode (#174) #596