mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
a3942b2d64
Summary: Basic ops to set/get/check/wait against a StoreHandler. Differential Revision: D4248059 fbshipit-source-id: cc53061fcc13823d4b9eed6b7c1c346b9e8ec991
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import uuid
|
|
|
|
from caffe2.distributed.store_ops_test_util import StoreOpsTests
|
|
from caffe2.python import core, workspace, dyndep
|
|
from caffe2.python.test_util import TestCase
|
|
|
|
dyndep.InitOpsLibrary("@/caffe2/caffe2/distributed:redis_store_handler_ops")
|
|
dyndep.InitOpsLibrary("@/caffe2/caffe2/distributed:store_ops")
|
|
|
|
|
|
class TestRedisStoreHandlerOp(TestCase):
|
|
def setUp(self):
|
|
super(TestRedisStoreHandlerOp, self).setUp()
|
|
self.uuid = str(uuid.uuid4()) + "/"
|
|
|
|
def tearDown(self):
|
|
super(TestRedisStoreHandlerOp, self).tearDown()
|
|
|
|
def create_store_handler(self):
|
|
store_handler = "store_handler"
|
|
workspace.RunOperatorOnce(
|
|
core.CreateOperator(
|
|
"RedisStoreHandlerCreate",
|
|
[],
|
|
[store_handler],
|
|
prefix=self.uuid,
|
|
host=os.getenv("REDIS_HOST", "localhost"),
|
|
port=int(os.getenv("REDIS_PORT", 6379))))
|
|
return store_handler
|
|
|
|
def test_set_get(self):
|
|
StoreOpsTests.test_set_get(self.create_store_handler)
|