Atletico Madrid are looking to finalise a deal for striker Diego Costa after a turbulent summer between the Spain international and Antonio Conte.