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

Tutorials nighly fix #16179

Merged
merged 13 commits into from
Sep 17, 2019
2 changes: 1 addition & 1 deletion docs/tutorials/gluon/custom_layer.md
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ class NormalizationHybridLayer(gluon.HybridBlock):

self.scales = self.params.get('scales',
shape=scales.shape,
init=mx.init.Constant(scales.asnumpy().tolist()), # Convert to regular list to make this object serializable
init=mx.init.Constant(scales.asnumpy()), # Convert to regular list to make this object serializable
differentiable=False)

def hybrid_forward(self, F, x, weights, scales):
Expand Down
3 changes: 3 additions & 0 deletions docs/tutorials/python/profiler.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ from mxnet import profiler

profiler.set_config(profile_all=True,
aggregate_stats=True,
continuous_dump=True,
filename='profile_output.json')
```

Expand Down Expand Up @@ -195,6 +196,8 @@ print(profiler.dumps())
You can also dump the information collected by the profiler into a `json` file using the `profiler.dump()` function and view it in a browser.

```python
profiler.set_state('run')
profiler.set_state('stop')
profiler.dump(finished=False)
```

Expand Down
2 changes: 1 addition & 1 deletion docs/tutorials/sparse/train_gluon.md
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ Similar to using a [`DataLoader`](https://mxnet.incubator.apache.org/versions/ma


```python
data_iter = mx.io.LibSVMIter(data_libsvm='test.libsvm', data_shape=(num_features,), label_shape=(1,), batch_size=10)
data_iter = mx.io.LibSVMIter(data_libsvm=filepath, data_shape=(num_features,), label_shape=(1,), batch_size=10)
for batch in data_iter:
data = batch.data[0]
print('data.stype: {}'.format(data.stype))
Expand Down
6 changes: 3 additions & 3 deletions tests/nightly/JenkinsfileForBinaries
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ core_logic: {
utils.pack_lib('gpu', mx_lib)
}
}
},
}/*,
Vikas-kum marked this conversation as resolved.
Show resolved Hide resolved
'CPU: USE_INT64_TENSOR_SIZE': {
node(NODE_LINUX_CPU) {
ws('workspace/build-cpu-int64') {
Expand All @@ -57,7 +57,7 @@ core_logic: {
utils.pack_lib('gpu_int64', mx_cmake_lib)
}
}
}
}*/
}

stage('NightlyTests'){
Expand All @@ -73,7 +73,7 @@ core_logic: {
node('mxnetlinux-gpu-p3-8xlarge') {
ws('workspace/nt-KVStoreTest') {
utils.unpack_and_init('gpu', mx_lib)
utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_KVStore_singleNode', true)
utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_KVStore_singleNode', true)
}
}
},
Expand Down
14 changes: 7 additions & 7 deletions tests/tutorials/test_tutorials.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,6 @@ def _test_tutorial_nb(tutorial):
"""
return run_notebook(tutorial, TUTORIAL_DIR, kernel=KERNEL, no_cache=NO_CACHE)


def test_basic_ndarray():
assert _test_tutorial_nb('basic/ndarray')

Expand Down Expand Up @@ -114,10 +113,10 @@ def test_gluon_save_load_params():

def test_gluon_hybrid():
assert _test_tutorial_nb('gluon/hybrid')

"""
Vikas-kum marked this conversation as resolved.
Show resolved Hide resolved
def test_gluon_performance():
assert _test_tutorial_nb('gluon/performance')
"""
def test_gluon_pretrained_models():
assert _test_tutorial_nb('gluon/pretrained_models')

Expand Down Expand Up @@ -183,10 +182,10 @@ def test_module_to_gluon():

def test_python_types_of_data_augmentation():
assert _test_tutorial_nb('python/types_of_data_augmentation')

"""
def test_python_profiler():
assert _test_tutorial_nb('python/profiler')

"""
def test_sparse_row_sparse():
assert _test_tutorial_nb('sparse/row_sparse')

Expand Down Expand Up @@ -216,6 +215,7 @@ def test_control_flow():

def test_amp():
assert _test_tutorial_nb('amp/amp_tutorial')

"""
def test_mkldnn_quantization():
assert _test_tutorial_nb('mkldnn/mkldnn_quantization')
assert _test_tutorial_nb('mkldnn/mkldnn_quantization')
"""
2 changes: 1 addition & 1 deletion tests/utils/notebook_test/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@


IPYTHON_VERSION = 4 # Pin to ipython version 4.
TIME_OUT = 10*60 # Maximum 10 mins/test. Reaching timeout causes test failure.
TIME_OUT = 15*60 # Maximum 10 mins/test. Reaching timeout causes test failure.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: can you update the comment to max 15 mins/ test

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i am going to merge this PR to unblock nightlies. @access2rohit since this change is small, would you be able to add this change to one of your WIP PRs?

ATTEMPTS = 8
KERNEL_ERROR_MSG = 'Kernel died before replying to kernel_info'

Expand Down