code: def test_reuse_of_env_thread_in_function_is_well_formed():
      """An env thread may be reused within a PrimFunc
  
      The `T.env_thread` has unique semantics, and may be defined at
      multiple locations without the TIR being considered ill-formed.
      """
  
      @T.prim_func
      def func(A: T.Buffer([256], "float32")):
          threadIdx_x = T.env_thread("threadIdx.x")
          with T.launch_thread(threadIdx_x, 256):
              A[threadIdx_x] = A[threadIdx_x] + 1.0
  
          with T.launch_thread(threadIdx_x, 256):
              A[threadIdx_x] = A[threadIdx_x] + 2.0
  
      tvm.tir.analysis.verify_well_formed(func)
  
file_path: tvm/tests/python/tir-analysis/test_tir_analysis_verify_well_formed.py
imports: import pytest
  import tvm
  import tvm.testing
  from tvm.script import ir, tir
label_desc: Use image node_id 7 for calling a global function with example usage: test_reuse_of_env_thread_in_function_is_well_formed() without return types