Update pair_wise_kernel.py

jimmytwei · web-flow · commit 2c7220852a36 · 2024-03-21T19:30:23.000-07:00
diff --git a/AI-and-Analytics/Jupyter/Numba_dpex_Essentials_training/03_dpex_Pairwise_Distance/lab/pair_wise_kernel.py b/AI-and-Analytics/Jupyter/Numba_dpex_Essentials_training/03_dpex_Pairwise_Distance/lab/pair_wise_kernel.py
@@ -22,7 +22,7 @@ def pairwise_python(X1, X2, D):
 
 
 def pw_distance(X1, X2, D):
-    pairwise_python[X1.shape[0],](X1, X2, D)
+    pairwise_python[nbdx.Range(X1.shape[0])](X1, X2, D) 
 
 
 base_pair_wise_gpu.run("Pairwise Distance Kernel", pw_distance)
