fix ci

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

python/tutorials/05-layer-norm.py

@@ -34,9 +34,6 @@
 import triton
 import triton.language as tl
 
-DEVICE = triton.runtime.driver.active.get_current_target().backend
-
-
 try:
     # This is https://github.com/NVIDIA/apex, NOT the apex on PyPi, so it
     # should not be added to extras_require in setup.py.
@@ -45,6 +42,8 @@
 except ModuleNotFoundError:
     HAS_APEX = False
 
+DEVICE = triton.runtime.driver.active.get_current_target().backend
+
 
 @triton.jit
 def _layer_norm_fwd_fused(