Merge pull request #9955 from tvegas1/go_serialized

GO/TESTS: Use serialized thread mode
openucx · Jun 17, 2024 · 0bda7b9 · 0bda7b9
2 parents 46fcc4b + 16ccaaf
commit 0bda7b9
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/bindings/go/src/ucx/ucs_constants.go b/bindings/go/src/ucx/ucs_constants.go
@@ -14,8 +14,9 @@ import "C"
 type UcsThreadMode int
 
 const (
-	UCS_THREAD_MODE_SINGLE UcsThreadMode = C.UCS_THREAD_MODE_SINGLE
-	UCS_THREAD_MODE_MULTI  UcsThreadMode = C.UCS_THREAD_MODE_MULTI
+	UCS_THREAD_MODE_SINGLE     UcsThreadMode = C.UCS_THREAD_MODE_SINGLE
+	UCS_THREAD_MODE_SERIALIZED UcsThreadMode = C.UCS_THREAD_MODE_SERIALIZED
+	UCS_THREAD_MODE_MULTI      UcsThreadMode = C.UCS_THREAD_MODE_MULTI
 )
 
 type UcsMemoryType int

diff --git a/bindings/go/tests/endpoint_test.go b/bindings/go/tests/endpoint_test.go
@@ -112,7 +112,7 @@ func TestUcpEpTag(t *testing.T) {
 		receiver := prepareContext(t, nil)
 		t.Logf("Testing tag %v -> %v", memType.senderMemType, memType.recvMemType)
 
-		ucpWorkerParams := (&UcpWorkerParams{}).SetThreadMode(UCS_THREAD_MODE_SINGLE)
+		ucpWorkerParams := (&UcpWorkerParams{}).SetThreadMode(UCS_THREAD_MODE_SERIALIZED)
 		ucpWorkerParams.WakeupTagSend().WakeupTagRecv()
 
 		receiver.worker, _ = receiver.context.NewWorker(ucpWorkerParams)

diff --git a/bindings/go/tests/listener_test.go b/bindings/go/tests/listener_test.go
@@ -19,7 +19,7 @@ func TestUcpListener(t *testing.T) {
 
 	ucpParams := (&UcpParams{}).EnableTag()
 
-	ucpWorkerParams := (&UcpWorkerParams{}).SetThreadMode(UCS_THREAD_MODE_SINGLE)
+	ucpWorkerParams := (&UcpWorkerParams{}).SetThreadMode(UCS_THREAD_MODE_SERIALIZED)
 
 	listenerParams := &UcpListenerParams{}
 	listenerParams.SetSocketAddress(addr)

diff --git a/bindings/go/tests/worker_test.go b/bindings/go/tests/worker_test.go
@@ -17,7 +17,7 @@ func TestUcpWorkerEfd(t *testing.T) {
 	ucpContext, _ := NewUcpContext((&UcpParams{}).EnableTag().EnableWakeup())
 	defer ucpContext.Close()
 	ucpWorkerParams := &UcpWorkerParams{}
-	ucpWorkerParams.SetThreadMode(UCS_THREAD_MODE_SINGLE)
+	ucpWorkerParams.SetThreadMode(UCS_THREAD_MODE_SERIALIZED)
 	ucpWorkerParams.WakeupEdge()
 
 	ucpWorker, err := ucpContext.NewWorker(ucpWorkerParams)