Drought-hypersensitive mutant 1 (DSM1) gene can be helpful. But a single gene cannot give expected result, a combination of such type of genes can be taken and pyramiding those gene can be more helpful in establishing drought resistance in a crop.
Just a suggestion, if you understand the genetics (basic and quantitative), you would have not asked that question. Any trait in any organism is not expressed solely on the basis of only one gene. There is always a complex interaction. If it is just one gene which can provide tolerance, they would have been cloned and propogated in all commercial / cultivated varieties for arid climate.
Totally agreed with Rajesh and Abhijeet, single gene could not provide drought / heat tolerance, since both trait were complex, number of genes/pathways were involved.
Of course, drought and heat tolerance both are controlled by complex of genes, but often there are one or several major genes and a lot of minor genes, which direct or indirect influence on the trait. A mutation in one of the major genes can significant change a trait. So, your question can be answered, such as it was done by Rajesh Dev Sarkar . You may search wheat samples with mutant alleles of genes of drought /heat tolerance or use ortological genes of landrases of wild spesies using interspecific crossing.