import _triton_alloc_setup