Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

CI test randomness 3 #9791

Merged
merged 12 commits into from
Feb 18, 2018
Merged
Prev Previous commit
Next Next commit
Merge branch 'fork_master' into ci_test_randomness3
  • Loading branch information
DickJC123 committed Feb 16, 2018
commit 9c6e3eb2384cfe82d69e06bf037c46f337b1ade2
8 changes: 8 additions & 0 deletions tests/python/gpu/test_operator_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -1111,6 +1111,14 @@ def test_activation_with_type():
check_consistency(sym, ctx_list)


@with_seed()
def test_lrn():
sym = mx.sym.LRN(alpha=0.0001, beta=0.75, knorm=2, nsize=5, name='lrn')
ctx_list = [{'ctx': mx.gpu(0), 'lrn_data': (2, 6, 10, 10), 'type_dict': {'lrn_data': np.float32}},
{'ctx': mx.cpu(0), 'lrn_data': (2, 6, 10, 10), 'type_dict': {'lrn_data': np.float32}}]
check_consistency(sym, ctx_list)


@with_seed()
def test_embedding_with_type():
def test_embedding_helper(data_types, weight_types, low_pad, high_pad):
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.