def test_add(self):
left_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_a')
right_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_b')
left_mat = rpt(left_locator)
right_mat = rpt(right_locator)
result = left_mat.add(right_mat)
def test_gpu_load(self):
left_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_a')
right_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_b')
left_mat = rpt(left_locator)
right_mat = rpt(right_locator)
print("[test_gpu_add]: +++___")
left_mat.gpu_load(right_mat)
def test_matmul(self):
left_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_a')
right_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_b')
left_mat = rpt(left_locator)
right_mat = rpt(right_locator)
print("[test cpu matmul ]: +++___")
left_mat.mat_mul(right_mat)
def test_scalar_mul(self):
store_locator = ErStoreLocator(store_type="levelDb",
namespace="ns",
name='mat_a')
original = rpt(store_locator)
result = original.scalar_mul(2)