Fix stopping before all workers had finished

Author: ncw

find_arctan_formulae.go

@@ -473,6 +473,7 @@ func groupToMapKey(s []uint32) string {
 
 // Find the possible Norms for the prime group and call PSLQ on them
 func processPrimeGroupsWorker(primeMap map[uint32][]*Norm, ch chan []uint32, wg *sync.WaitGroup) {
+	defer wg.Done()
 	baseValid := make([]*Norm, 0, 100)
 	for group := range ch {
 		valid := baseValid[:0]
@@ -570,6 +571,7 @@ func processPrimeGroups(norms []*Norm, primeMap map[uint32][]*Norm, primesToProc
 	ch := make(chan []uint32, workers)
 
 	for range workers {
+		wg.Add(1)
 		go processPrimeGroupsWorker(primeMap, ch, &wg)
 	}
 
@@ -610,6 +612,7 @@ func processPrimeGroups(norms []*Norm, primeMap map[uint32][]*Norm, primesToProc
 	}
 
 	close(ch)
+	Printf("Waiting for workers to finish\n")
 	wg.Wait()
 
 	statsMu.Lock()