Makefile for doc_nccl and minor fix in nccl tutorial

Author: tharittk

Makefile

@@ -55,6 +55,13 @@ doc:
 	rm -rf source/tutorials && rm -rf build &&\
 	cd .. && sphinx-build -b html docs/source docs/build
 
+doc_nccl:
+	cp tutorials_nccl/* tutorials/
+	cd docs  && rm -rf source/api/generated && rm -rf source/gallery &&\
+	rm -rf source/tutorials && rm -rf source/tutorials && rm -rf build &&\
+	cd .. && sphinx-build -b html docs/source docs/build
+	rm tutorials/*_nccl.py
+
 docupdate:
 	cd docs && make html && cd ..
 

tutorials_nccl/poststack_nccl.py

@@ -15,8 +15,6 @@
 from pylops.avo.poststack import PoststackLinearModelling
 
 import pylops_mpi
-import pylops_mpi.utils
-import pylops_mpi.utils._nccl
 
 ###############################################################################
 # NCCL communication can be easily initialized with
@@ -198,5 +196,3 @@
     axs[5][2].imshow(minv3d_reg[:, :, 220].T.get(), cmap='gist_rainbow', vmin=m.min(), vmax=m.max())
     axs[5][2].set_title('Regularized Inverted Model iter x-y')
     axs[5][2].axis('tight')
-
-    plt.savefig("./poststack_inv_nccl.png")