BackgroundPlant architecture associated with increased grain yield and adaptation to the local environments is selected during wheat (Triticum aestivum) breeding. The internode length of individual stems and tiller length of individual plants are important for the determination of plant architecture. However, few studies have explored the genetic basis of these traits.ResultsHere, we conduct a genome-wide association study (GWAS) to dissect the genetic basis of geographical differentiation of these traits in 306 worldwide wheat accessions including both landraces and traditional varieties. We determine the changes of haplotypes for the associated genomic regions in frequency in 831 wheat accessions that are either introduced from other countries or developed in China from last two decades. We identify 83 loci that are associated with one trait, while the remaining 247 loci are pleiotropic. We also find 163 associated loci are under strong selective sweep. GWAS results demonstrate independent regulation of internode length of individual stems and consistent regulation of tiller length of individual plants. This makes it possible to obtain ideal haplotype combinations of the length of four internodes. We also find that the geographical distribution of the haplotypes explains the observed differences in internode length among the worldwide wheat accessions.ConclusionThis study provides insights into the genetic basis of plant architecture. It will facilitate gene functional analysis and molecular design of plant architecture for breeding.Supplementary InformationThe online version contains supplementary material available at 10.1186/s13059-023-02932-x.