Adding flex_attention benchmark for eager and compile mode

Adding flex_attention benchmark for eager and compile mode #594