forking bitburner write thread function

driver-avalon.c

@@ -923,12 +923,15 @@ static void *avalon_get_results(void *userdata)
 
 static void avalon_rotate_array(struct cgpu_info *avalon)
 {
-	int array_size = AVALON_ARRAY_SIZE;
+	avalon->queued = 0;
+	if (++avalon->work_array >= AVALON_ARRAY_SIZE)
+		avalon->work_array = 0;
+}
 
-	if (usb_ident(avalon) == IDENT_BTB)
+static void bitburner_rotate_array(struct cgpu_info *avalon)
-		array_size = BITBURNER_ARRAY_SIZE;
+{
 	avalon->queued = 0;
-	if (++avalon->work_array >= array_size)
+	if (++avalon->work_array >= BITBURNER_ARRAY_SIZE)
 		avalon->work_array = 0;
 }
 
@@ -979,24 +982,8 @@ static void avalon_reset_auto(struct avalon_info *info)
 	info->auto_hw = 0;
 }
 
-static void *avalon_send_tasks(void *userdata)
+static void avalon_adjust_freq(struct avalon_info *info, struct cgpu_info *avalon)
 {
-	struct cgpu_info *avalon = (struct cgpu_info *)userdata;
-	struct avalon_info *info = avalon->device_data;
-	const int avalon_get_work_count = info->miner_count;
-	char threadname[24];
-
-	snprintf(threadname, 24, "ava_send/%d", avalon->device_id);
-	RenameThread(threadname);
-
-	while (likely(!avalon->shutdown)) {
-		int start_count, end_count, i, j, ret;
-		struct avalon_task at;
-		bool idled = false;
-
-		while (avalon_buffer_full(avalon))
-			nmsleep(40);
-
 	if (opt_avalon_auto && info->auto_queued >= AVALON_AUTO_CYCLE) {
 		mutex_lock(&info->lock);
 		if (!info->optimal) {
@@ -1019,15 +1006,27 @@ static void *avalon_send_tasks(void *userdata)
 		avalon_reset_auto(info);
 		mutex_unlock(&info->lock);
 	}
+}
 
-		if (usb_ident(avalon) == IDENT_BTB) {
+static void *avalon_send_tasks(void *userdata)
-			// Give other threads the chance to acquire qlock
+{
-			i = 0;
+	struct cgpu_info *avalon = (struct cgpu_info *)userdata;
-			do {
+	struct avalon_info *info = avalon->device_data;
+	const int avalon_get_work_count = info->miner_count;
+	char threadname[24];
+
+	snprintf(threadname, 24, "ava_send/%d", avalon->device_id);
+	RenameThread(threadname);
+
+	while (likely(!avalon->shutdown)) {
+		int start_count, end_count, i, j, ret;
+		struct avalon_task at;
+		bool idled = false;
+
+		while (avalon_buffer_full(avalon))
 			nmsleep(40);
-			} while (!avalon->shutdown && i++ < 15
+
-				&& avalon->queued < avalon_get_work_count);
+		avalon_adjust_freq(info, avalon);
-		}
 
 		mutex_lock(&info->qlock);
 		start_count = avalon->work_array * avalon_get_work_count;
@@ -1037,13 +1036,87 @@ static void *avalon_send_tasks(void *userdata)
 				applog(LOG_INFO,
 				       "%s%i: Buffer full after only %d of %d work queued",
 					avalon->drv->name, avalon->device_id, j, avalon_get_work_count);
+				break;
+			}
+
+			if (likely(j < avalon->queued && !info->overheat && avalon->works[i])) {
+				avalon_init_task(&at, 0, 0, info->fan_pwm,
+						info->timeout, info->asic_count,
+						info->miner_count, 1, 0, info->frequency);
+				avalon_create_task(&at, avalon->works[i]);
+				info->auto_queued++;
+			} else {
+				int idle_freq = info->frequency;
 
-					if (usb_ident(avalon) != IDENT_BTB)
+				if (!info->idle++)
+					idled = true;
+				if (unlikely(info->overheat && opt_avalon_auto))
+					idle_freq = AVALON_MIN_FREQUENCY;
+				avalon_init_task(&at, 0, 0, info->fan_pwm,
+						info->timeout, info->asic_count,
+						info->miner_count, 1, 1, idle_freq);
+				/* Reset the auto_queued count if we end up
+				 * idling any miners. */
+				avalon_reset_auto(info);
+			}
+
+			ret = avalon_send_task(&at, avalon);
+
+			if (unlikely(ret == AVA_SEND_ERROR)) {
+				applog(LOG_ERR, "%s%i: Comms error(buffer)",
+				       avalon->drv->name, avalon->device_id);
+				dev_error(avalon, REASON_DEV_COMMS_ERROR);
+				info->reset = true;
 				break;
-					else {
+			}
+		}
+
+		avalon_rotate_array(avalon);
+		pthread_cond_signal(&info->qcond);
+		mutex_unlock(&info->qlock);
+
+		if (unlikely(idled)) {
+			applog(LOG_WARNING, "%s%i: Idled %d miners",
+			       avalon->drv->name, avalon->device_id, idled);
+		}
+	}
+	return NULL;
+}
+
+static void *bitburner_send_tasks(void *userdata)
+{
+	struct cgpu_info *avalon = (struct cgpu_info *)userdata;
+	struct avalon_info *info = avalon->device_data;
+	const int avalon_get_work_count = info->miner_count;
+	char threadname[24];
+
+	snprintf(threadname, 24, "ava_send/%d", avalon->device_id);
+	RenameThread(threadname);
+
+	while (likely(!avalon->shutdown)) {
+		int start_count, end_count, i, j, ret;
+		struct avalon_task at;
+		bool idled = false;
+
+		while (avalon_buffer_full(avalon))
+			nmsleep(40);
+
+		avalon_adjust_freq(info, avalon);
+
+		/* Give other threads a chance to acquire qlock. */
+		i = 0;
+		do {
+			nmsleep(40);
+		} while (!avalon->shutdown && i++ < 15
+			&& avalon->queued < avalon_get_work_count);
+
+		mutex_lock(&info->qlock);
+		start_count = avalon->work_array * avalon_get_work_count;
+		end_count = start_count + avalon_get_work_count;
+		for (i = start_count, j = 0; i < end_count; i++, j++) {
+			if (avalon_buffer_full(avalon)) {
 				while (avalon_buffer_full(avalon))
 					nmsleep(40);
-					}
 			}
 
 			if (likely(j < avalon->queued && !info->overheat && avalon->works[i])) {
@@ -1078,7 +1151,7 @@ static void *avalon_send_tasks(void *userdata)
 			}
 		}
 
-		avalon_rotate_array(avalon);
+		bitburner_rotate_array(avalon);
 		pthread_cond_signal(&info->qcond);
 		mutex_unlock(&info->qlock);
 
@@ -1095,9 +1168,13 @@ static bool avalon_prepare(struct thr_info *thr)
 	struct cgpu_info *avalon = thr->cgpu;
 	struct avalon_info *info = avalon->device_data;
 	int array_size = AVALON_ARRAY_SIZE;
+	void *(*write_thread_fn)(void *) = avalon_send_tasks;
 
-	if (usb_ident(avalon) == IDENT_BTB)
+	if (usb_ident(avalon) == IDENT_BTB) {
 		array_size = BITBURNER_ARRAY_SIZE;
+		write_thread_fn = bitburner_send_tasks;
+	}
+
 	free(avalon->works);
 	avalon->works = calloc(info->miner_count * sizeof(struct work *),
 			       array_size);
@@ -1113,7 +1190,7 @@ static bool avalon_prepare(struct thr_info *thr)
 	if (pthread_create(&info->read_thr, NULL, avalon_get_results, (void *)avalon))
 		quit(1, "Failed to create avalon read_thr");
 
-	if (pthread_create(&info->write_thr, NULL, avalon_send_tasks, (void *)avalon))
+	if (pthread_create(&info->write_thr, NULL, write_thread_fn, (void *)avalon))
 		quit(1, "Failed to create avalon write_thr");
 
 	avalon_init(avalon);