; LLVM IR from NVIDIA/DALI :: dali/operators/util (axis_args)

; Unoptimized IR (clang -disable-llvm-passes):
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"

%"struct.dali::ArgShapeFromSize" = type { i8 }
%"struct.dali::TensorListShape" = type { %"struct.dali::TensorListShapeBase.base", [4 x i8] }
%"struct.dali::TensorListShapeBase.base" = type <{ %"class.std::vector", i32 }>
%"class.std::vector" = type { %"struct.std::_Vector_base" }
%"struct.std::_Vector_base" = type { %"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl" }
%"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl" = type { %"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl_data" }
%"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl_data" = type { ptr, ptr, ptr }
%"struct.dali::TensorShape" = type { %"struct.dali::TensorShapeBase" }
%"struct.dali::TensorShapeBase" = type { %"class.dali::DeviceArray" }
%"class.dali::DeviceArray" = type { [1 x i64] }
%"class.dali::span" = type { ptr }
%"class.dali::AxisArgs" = type { i32, %"class.std::unique_ptr", i8, i8, %"class.dali::TensorLayout", %"class.dali::SmallVector", %"struct.dali::TensorListShape" }
%"class.std::unique_ptr" = type { %"struct.std::__uniq_ptr_data" }
%"struct.std::__uniq_ptr_data" = type { %"class.std::__uniq_ptr_impl" }
%"class.std::__uniq_ptr_impl" = type { %"class.std::tuple" }
%"class.std::tuple" = type { %"struct.std::_Tuple_impl" }
%"struct.std::_Tuple_impl" = type { %"struct.std::_Head_base.1" }
%"struct.std::_Head_base.1" = type { ptr }
%"class.dali::TensorLayout" = type { [16 x i8] }
%"class.dali::SmallVector" = type { %union.anon, i64 }
%union.anon = type { %struct.anon, [8 x i8] }
%struct.anon = type { ptr, i64 }
%"struct.dali::TensorListViewBase" = type { %"struct.dali::TensorListShape", %"class.std::vector.40" }
%"class.std::vector.40" = type { %"struct.std::_Vector_base.41" }
%"struct.std::_Vector_base.41" = type { %"struct.std::_Vector_base<const int *, std::allocator<const int *>>::_Vector_impl" }
%"struct.std::_Vector_base<const int *, std::allocator<const int *>>::_Vector_impl" = type { %"struct.std::_Vector_base<const int *, std::allocator<const int *>>::_Vector_impl_data" }
%"struct.std::_Vector_base<const int *, std::allocator<const int *>>::_Vector_impl_data" = type { ptr, ptr, ptr }

@.str.9 = external hidden unnamed_addr constant [6 x i8], align 1
@.str.10 = external hidden unnamed_addr constant [78 x i8], align 1
@__PRETTY_FUNCTION__._ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii = external hidden unnamed_addr constant [82 x i8], align 1

declare i32 @__gxx_personality_v0(...)

; Function Attrs: nocallback nofree nosync nounwind willreturn memory(argmem: readwrite)
declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture) #0

; Function Attrs: nocallback nofree nosync nounwind willreturn memory(argmem: readwrite)
declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture) #0

; Function Attrs: inlinehint mustprogress nounwind uwtable
declare dso_local void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28)) unnamed_addr #1 align 2

; Function Attrs: mustprogress uwtable
define dso_local void @_ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii(ptr noundef nonnull align 8 dereferenceable(104) %0, ptr noundef nonnull align 8 dereferenceable(328) %1, ptr noundef nonnull align 8 dereferenceable(256) %2, i32 noundef %3, i32 noundef %4) #2 align 2 personality ptr @__gxx_personality_v0 {
  %6 = alloca ptr, align 8
  %7 = alloca ptr, align 8
  %8 = alloca ptr, align 8
  %9 = alloca i32, align 4
  %10 = alloca i32, align 4
  %11 = alloca %"struct.dali::ArgShapeFromSize", align 1
  %12 = alloca %"struct.dali::TensorListShape", align 8
  %13 = alloca %"struct.dali::TensorShape", align 8
  %14 = alloca ptr, align 8
  %15 = alloca i32, align 4
  %16 = alloca %"struct.dali::TensorListShape", align 8
  %17 = alloca %"struct.dali::TensorShape", align 8
  %18 = alloca %"struct.dali::TensorShape", align 8
  %19 = alloca %"struct.dali::TensorListShape", align 8
  %20 = alloca i32, align 4
  %21 = alloca %"class.dali::span", align 8
  store ptr %0, ptr %6, align 8, !tbaa !5
  store ptr %1, ptr %7, align 8, !tbaa !5
  store ptr %2, ptr %8, align 8, !tbaa !5
  store i32 %3, ptr %9, align 4, !tbaa !9
  store i32 %4, ptr %10, align 4, !tbaa !9
  %22 = load ptr, ptr %6, align 8
  %23 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 3
  %24 = load i8, ptr %23, align 1, !tbaa !11, !range !29, !noundef !30
  %25 = trunc i8 %24 to i1
  br i1 %25, label %26, label %44

26:                                               ; preds = %5
  %27 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 1
  %28 = call noundef zeroext i1 @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEcvbEv(ptr noundef nonnull align 8 dereferenceable(8) %27) #6
  br i1 %28, label %29, label %30

29:                                               ; preds = %26
  br label %32

30:                                               ; preds = %26
  call void @__assert_fail(ptr noundef @.str.9, ptr noundef @.str.10, i32 noundef 57, ptr noundef @__PRETTY_FUNCTION__._ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii) #7
  unreachable

31:                                               ; No predecessors!
  br label %32

32:                                               ; preds = %31, %29
  %33 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 1
  %34 = call noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8) %33) #6
  %35 = load ptr, ptr %7, align 8, !tbaa !5
  %36 = load ptr, ptr %8, align 8, !tbaa !5
  %37 = load i32, ptr %9, align 4, !tbaa !9
  call void @llvm.lifetime.start.p0(i64 1, ptr %11) #6
  call void @_ZN4dali8ArgValueIiLi1EE7AcquireINS_16ArgShapeFromSizeILi1EEEEEvRKNS_6OpSpecERKNS_17ArgumentWorkspaceEiNS_13ArgValueFlagsEOT_(ptr noundef nonnull align 8 dereferenceable(129) %34, ptr noundef nonnull align 8 dereferenceable(328) %35, ptr noundef nonnull align 8 dereferenceable(256) %36, i32 noundef %37, i32 noundef 0, ptr noundef nonnull align 1 dereferenceable(1) %11)
  call void @llvm.lifetime.end.p0(i64 1, ptr %11) #6
  %38 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 1
  %39 = call noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8) %38) #6
  %40 = call noundef nonnull align 8 dereferenceable(56) ptr @_ZNK4dali8ArgValueIiLi1EE3getEv(ptr noundef nonnull align 8 dereferenceable(129) %39)
  %41 = getelementptr inbounds %"struct.dali::TensorListViewBase", ptr %40, i32 0, i32 0
  %42 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %43 = call noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSERKS1_(ptr noundef nonnull align 8 dereferenceable(28) %42, ptr noundef nonnull align 8 dereferenceable(28) %41)
  br label %72

44:                                               ; preds = %5
  %45 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 2
  %46 = load i8, ptr %45, align 8, !tbaa !31, !range !29, !noundef !30
  %47 = trunc i8 %46 to i1
  br i1 %47, label %48, label %60

48:                                               ; preds = %44
  call void @llvm.lifetime.start.p0(i64 32, ptr %12) #6
  %49 = load i32, ptr %9, align 4, !tbaa !9
  call void @llvm.lifetime.start.p0(i64 8, ptr %13) #6
  %50 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 4
  %51 = call noundef zeroext i8 @_ZNK4dali12TensorLayout4sizeEv(ptr noundef nonnull align 1 dereferenceable(16) %50) #6
  %52 = zext i8 %51 to i64
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %13, i64 noundef %52)
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind writable sret(%"struct.dali::TensorListShape") align 8 %12, i32 noundef %49, ptr noundef nonnull align 8 dereferenceable(8) %13)
  %53 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %54 = invoke noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %53, ptr noundef nonnull align 8 dereferenceable(28) %12)
          to label %55 unwind label %56

55:                                               ; preds = %48
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %12) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr %13) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %12) #6
  br label %71

56:                                               ; preds = %48
  %57 = landingpad { ptr, i32 }
          cleanup
  %58 = extractvalue { ptr, i32 } %57, 0
  store ptr %58, ptr %14, align 8
  %59 = extractvalue { ptr, i32 } %57, 1
  store i32 %59, ptr %15, align 4
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %12) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr %13) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %12) #6
  br label %118

60:                                               ; preds = %44
  call void @llvm.lifetime.start.p0(i64 32, ptr %16) #6
  %61 = load i32, ptr %9, align 4, !tbaa !9
  call void @llvm.lifetime.start.p0(i64 8, ptr %17) #6
  %62 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 5
  %63 = call noundef i64 @_ZNK4dali11SmallVectorIiLm6ESaIiEE4sizeEv(ptr noundef nonnull align 8 dereferenceable(32) %62)
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %17, i64 noundef %63)
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind writable sret(%"struct.dali::TensorListShape") align 8 %16, i32 noundef %61, ptr noundef nonnull align 8 dereferenceable(8) %17)
  %64 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %65 = invoke noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %64, ptr noundef nonnull align 8 dereferenceable(28) %16)
          to label %66 unwind label %67

66:                                               ; preds = %60
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %16) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr %17) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %16) #6
  br label %71

67:                                               ; preds = %60
  %68 = landingpad { ptr, i32 }
          cleanup
  %69 = extractvalue { ptr, i32 } %68, 0
  store ptr %69, ptr %14, align 8
  %70 = extractvalue { ptr, i32 } %68, 1
  store i32 %70, ptr %15, align 4
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %16) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr %17) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %16) #6
  br label %118

71:                                               ; preds = %66, %55
  br label %72

72:                                               ; preds = %71, %32
  %73 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 0
  %74 = load i32, ptr %73, align 8, !tbaa !32
  %75 = and i32 %74, 4
  %76 = icmp ne i32 %75, 0
  br i1 %76, label %77, label %117

77:                                               ; preds = %72
  call void @llvm.lifetime.start.p0(i64 8, ptr %18) #6
  %78 = load i32, ptr %10, align 4, !tbaa !9
  %79 = sext i32 %78 to i64
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %18, i64 noundef %79)
  %80 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %81 = call noundef i64 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE12num_elementsEv(ptr noundef nonnull align 8 dereferenceable(28) %80)
  %82 = icmp eq i64 %81, 0
  br i1 %82, label %83, label %92

83:                                               ; preds = %77
  call void @llvm.lifetime.start.p0(i64 32, ptr %19) #6
  %84 = load i32, ptr %9, align 4, !tbaa !9
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind writable sret(%"struct.dali::TensorListShape") align 8 %19, i32 noundef %84, ptr noundef nonnull align 8 dereferenceable(8) %18)
  %85 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %86 = invoke noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %85, ptr noundef nonnull align 8 dereferenceable(28) %19)
          to label %87 unwind label %88

87:                                               ; preds = %83
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %19) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %19) #6
  br label %116

88:                                               ; preds = %83
  %89 = landingpad { ptr, i32 }
          cleanup
  %90 = extractvalue { ptr, i32 } %89, 0
  store ptr %90, ptr %14, align 8
  %91 = extractvalue { ptr, i32 } %89, 1
  store i32 %91, ptr %15, align 4
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %19) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr %19) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr %18) #6
  br label %118

92:                                               ; preds = %77
  call void @llvm.lifetime.start.p0(i64 4, ptr %20) #6
  store i32 0, ptr %20, align 4, !tbaa !9
  br label %93

93:                                               ; preds = %112, %92
  %94 = load i32, ptr %20, align 4, !tbaa !9
  %95 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %96 = call noundef i32 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE4sizeEv(ptr noundef nonnull align 8 dereferenceable(28) %95)
  %97 = icmp slt i32 %94, %96
  br i1 %97, label %99, label %98

98:                                               ; preds = %93
  call void @llvm.lifetime.end.p0(i64 4, ptr %20) #6
  br label %115

99:                                               ; preds = %93
  call void @llvm.lifetime.start.p0(i64 8, ptr %21) #6
  %100 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %101 = load i32, ptr %20, align 4, !tbaa !9
  %102 = sext i32 %101 to i64
  %103 = call ptr @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE17tensor_shape_spanEl(ptr noundef nonnull align 8 dereferenceable(28) %100, i64 noundef %102)
  %104 = getelementptr inbounds %"class.dali::span", ptr %21, i32 0, i32 0
  store ptr %103, ptr %104, align 8
  %105 = call noundef i64 @_ZN4dali6volumeINS_4spanIlLl1EEEEEDTcl6volumeclsr4daliE5beginfp_Eclsr4daliE3endfp_EEERKT_(ptr noundef nonnull align 8 dereferenceable(8) %21)
  %106 = icmp eq i64 %105, 0
  call void @llvm.lifetime.end.p0(i64 8, ptr %21) #6
  br i1 %106, label %107, label %111

107:                                              ; preds = %99
  %108 = getelementptr inbounds %"class.dali::AxisArgs", ptr %22, i32 0, i32 6
  %109 = load i32, ptr %20, align 4, !tbaa !9
  %110 = sext i32 %109 to i64
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE16set_tensor_shapeElRKNS_11TensorShapeILi1EEE(ptr noundef nonnull align 8 dereferenceable(28) %108, i64 noundef %110, ptr noundef nonnull align 8 dereferenceable(8) %18)
  br label %111

111:                                              ; preds = %107, %99
  br label %112

112:                                              ; preds = %111
  %113 = load i32, ptr %20, align 4, !tbaa !9
  %114 = add nsw i32 %113, 1
  store i32 %114, ptr %20, align 4, !tbaa !9
  br label %93, !llvm.loop !33

115:                                              ; preds = %98
  br label %116

116:                                              ; preds = %115, %87
  call void @llvm.lifetime.end.p0(i64 8, ptr %18) #6
  br label %117

117:                                              ; preds = %116, %72
  ret void

118:                                              ; preds = %88, %67, %56
  %119 = load ptr, ptr %14, align 8
  %120 = load i32, ptr %15, align 4
  %121 = insertvalue { ptr, i32 } poison, ptr %119, 0
  %122 = insertvalue { ptr, i32 } %121, i32 %120, 1
  resume { ptr, i32 } %122
}

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef zeroext i1 @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEcvbEv(ptr noundef nonnull align 8 dereferenceable(8)) #3 align 2

; Function Attrs: noreturn nounwind
declare void @__assert_fail(ptr noundef, ptr noundef, i32 noundef, ptr noundef) #4

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8)) #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali8ArgValueIiLi1EE7AcquireINS_16ArgShapeFromSizeILi1EEEEEvRKNS_6OpSpecERKNS_17ArgumentWorkspaceEiNS_13ArgValueFlagsEOT_(ptr noundef nonnull align 8 dereferenceable(129), ptr noundef nonnull align 8 dereferenceable(328), ptr noundef nonnull align 8 dereferenceable(256), i32 noundef, i32 noundef, ptr noundef nonnull align 1 dereferenceable(1)) #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef nonnull align 8 dereferenceable(56) ptr @_ZNK4dali8ArgValueIiLi1EE3getEv(ptr noundef nonnull align 8 dereferenceable(129)) #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSERKS1_(ptr noundef nonnull align 8 dereferenceable(28), ptr noundef nonnull align 8 dereferenceable(28)) #2 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind noalias writable sret(%"struct.dali::TensorListShape") align 8, i32 noundef, ptr noundef nonnull align 8 dereferenceable(8)) #2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef zeroext i8 @_ZNK4dali12TensorLayout4sizeEv(ptr noundef nonnull align 1 dereferenceable(16)) #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8), i64 noundef) unnamed_addr #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28), ptr noundef nonnull align 8 dereferenceable(28)) #3 align 2

; Function Attrs: inlinehint mustprogress nounwind uwtable
declare dso_local noundef i64 @_ZNK4dali11SmallVectorIiLm6ESaIiEE4sizeEv(ptr noundef nonnull align 8 dereferenceable(32)) #1 align 2

; Function Attrs: mustprogress uwtable
declare dso_local noundef i64 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE12num_elementsEv(ptr noundef nonnull align 8 dereferenceable(28)) #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef i32 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE4sizeEv(ptr noundef nonnull align 8 dereferenceable(28)) #3 align 2

; Function Attrs: inlinehint mustprogress uwtable
declare dso_local noundef i64 @_ZN4dali6volumeINS_4spanIlLl1EEEEEDTcl6volumeclsr4daliE5beginfp_Eclsr4daliE3endfp_EEERKT_(ptr noundef nonnull align 8 dereferenceable(8)) #5

; Function Attrs: mustprogress uwtable
declare dso_local ptr @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE17tensor_shape_spanEl(ptr noundef nonnull align 8 dereferenceable(28), i64 noundef) #2 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE16set_tensor_shapeElRKNS_11TensorShapeILi1EEE(ptr noundef nonnull align 8 dereferenceable(28), i64 noundef, ptr noundef nonnull align 8 dereferenceable(8)) #2 align 2

attributes #0 = { nocallback nofree nosync nounwind willreturn memory(argmem: readwrite) }
attributes #1 = { inlinehint mustprogress nounwind uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #2 = { mustprogress uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #3 = { mustprogress nounwind uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #4 = { noreturn nounwind "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #5 = { inlinehint mustprogress uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #6 = { nounwind }
attributes #7 = { noreturn nounwind }

!llvm.module.flags = !{!0, !1, !2, !3}
!llvm.ident = !{!4}

!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{i32 8, !"PIC Level", i32 2}
!2 = !{i32 7, !"PIE Level", i32 2}
!3 = !{i32 7, !"uwtable", i32 2}
!4 = !{!"clang version 19.1.0 (git@github.com:llvm/llvm-project.git a4bf6cd7cfb1a1421ba92bca9d017b49936c55e4)"}
!5 = !{!6, !6, i64 0}
!6 = !{!"any pointer", !7, i64 0}
!7 = !{!"omnipotent char", !8, i64 0}
!8 = !{!"Simple C++ TBAA"}
!9 = !{!10, !10, i64 0}
!10 = !{!"int", !7, i64 0}
!11 = !{!12, !19, i64 17}
!12 = !{!"_ZTSN4dali8AxisArgsE", !10, i64 0, !13, i64 8, !19, i64 16, !19, i64 17, !20, i64 18, !21, i64 40, !23, i64 72}
!13 = !{!"_ZTSSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EE", !14, i64 0}
!14 = !{!"_ZTSSt15__uniq_ptr_dataIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_ELb1ELb1EE", !15, i64 0}
!15 = !{!"_ZTSSt15__uniq_ptr_implIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EE", !16, i64 0}
!16 = !{!"_ZTSSt5tupleIJPN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEE", !17, i64 0}
!17 = !{!"_ZTSSt11_Tuple_implILm0EJPN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEE", !18, i64 0}
!18 = !{!"_ZTSSt10_Head_baseILm0EPN4dali8ArgValueIiLi1EEELb0EE", !6, i64 0}
!19 = !{!"bool", !7, i64 0}
!20 = !{!"_ZTSN4dali12TensorLayoutE", !7, i64 0}
!21 = !{!"_ZTSN4dali11SmallVectorIiLm6ESaIiEEE", !7, i64 0, !22, i64 24}
!22 = !{!"long", !7, i64 0}
!23 = !{!"_ZTSN4dali15TensorListShapeILi1EEE", !24, i64 0}
!24 = !{!"_ZTSN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EEE", !25, i64 0, !10, i64 24}
!25 = !{!"_ZTSSt6vectorIlSaIlEE", !26, i64 0}
!26 = !{!"_ZTSSt12_Vector_baseIlSaIlEE", !27, i64 0}
!27 = !{!"_ZTSNSt12_Vector_baseIlSaIlEE12_Vector_implE", !28, i64 0}
!28 = !{!"_ZTSNSt12_Vector_baseIlSaIlEE17_Vector_impl_dataE", !6, i64 0, !6, i64 8, !6, i64 16}
!29 = !{i8 0, i8 2}
!30 = !{}
!31 = !{!12, !19, i64 16}
!32 = !{!12, !10, i64 0}
!33 = distinct !{!33, !34}
!34 = !{!"llvm.loop.mustprogress"}


; Optimized IR (opt -passes='default<O3>'):
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"

%"struct.dali::ArgShapeFromSize" = type { i8 }
%"struct.dali::TensorListShape" = type { %"struct.dali::TensorListShapeBase.base", [4 x i8] }
%"struct.dali::TensorListShapeBase.base" = type <{ %"class.std::vector", i32 }>
%"class.std::vector" = type { %"struct.std::_Vector_base" }
%"struct.std::_Vector_base" = type { %"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl" }
%"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl" = type { %"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl_data" }
%"struct.std::_Vector_base<long, std::allocator<long>>::_Vector_impl_data" = type { ptr, ptr, ptr }
%"struct.dali::TensorShape" = type { %"struct.dali::TensorShapeBase" }
%"struct.dali::TensorShapeBase" = type { %"class.dali::DeviceArray" }
%"class.dali::DeviceArray" = type { [1 x i64] }
%"class.dali::span" = type { ptr }

@.str.9 = external hidden unnamed_addr constant [6 x i8], align 1
@.str.10 = external hidden unnamed_addr constant [78 x i8], align 1
@__PRETTY_FUNCTION__._ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii = external hidden unnamed_addr constant [82 x i8], align 1

declare i32 @__gxx_personality_v0(...)

; Function Attrs: mustprogress nocallback nofree nosync nounwind willreturn memory(argmem: readwrite)
declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture) #0

; Function Attrs: mustprogress nocallback nofree nosync nounwind willreturn memory(argmem: readwrite)
declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture) #0

; Function Attrs: inlinehint mustprogress nounwind uwtable
declare dso_local void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28)) unnamed_addr #1 align 2

; Function Attrs: mustprogress uwtable
define dso_local void @_ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii(ptr noundef nonnull align 8 dereferenceable(104) %0, ptr noundef nonnull align 8 dereferenceable(328) %1, ptr noundef nonnull align 8 dereferenceable(256) %2, i32 noundef %3, i32 noundef %4) local_unnamed_addr #2 align 2 personality ptr @__gxx_personality_v0 {
  %6 = alloca %"struct.dali::ArgShapeFromSize", align 1
  %7 = alloca %"struct.dali::TensorListShape", align 8
  %8 = alloca %"struct.dali::TensorShape", align 8
  %9 = alloca %"struct.dali::TensorListShape", align 8
  %10 = alloca %"struct.dali::TensorShape", align 8
  %11 = alloca %"struct.dali::TensorShape", align 8
  %12 = alloca %"struct.dali::TensorListShape", align 8
  %13 = alloca %"class.dali::span", align 8
  %14 = getelementptr inbounds i8, ptr %0, i64 17
  %15 = load i8, ptr %14, align 1, !tbaa !5, !range !27, !noundef !28
  %16 = trunc nuw i8 %15 to i1
  br i1 %16, label %17, label %27

17:                                               ; preds = %5
  %18 = getelementptr inbounds i8, ptr %0, i64 8
  %19 = tail call noundef zeroext i1 @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEcvbEv(ptr noundef nonnull align 8 dereferenceable(8) %18) #6
  br i1 %19, label %21, label %20

20:                                               ; preds = %17
  tail call void @__assert_fail(ptr noundef nonnull @.str.9, ptr noundef nonnull @.str.10, i32 noundef 57, ptr noundef nonnull @__PRETTY_FUNCTION__._ZN4dali8AxisArgs7AcquireERKNS_6OpSpecERKNS_17ArgumentWorkspaceEii) #7
  unreachable

21:                                               ; preds = %17
  %22 = tail call noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8) %18) #6
  call void @llvm.lifetime.start.p0(i64 1, ptr nonnull %6) #6
  call void @_ZN4dali8ArgValueIiLi1EE7AcquireINS_16ArgShapeFromSizeILi1EEEEEvRKNS_6OpSpecERKNS_17ArgumentWorkspaceEiNS_13ArgValueFlagsEOT_(ptr noundef nonnull align 8 dereferenceable(129) %22, ptr noundef nonnull align 8 dereferenceable(328) %1, ptr noundef nonnull align 8 dereferenceable(256) %2, i32 noundef %3, i32 noundef 0, ptr noundef nonnull align 1 dereferenceable(1) %6)
  call void @llvm.lifetime.end.p0(i64 1, ptr nonnull %6) #6
  %23 = call noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8) %18) #6
  %24 = call noundef nonnull align 8 dereferenceable(56) ptr @_ZNK4dali8ArgValueIiLi1EE3getEv(ptr noundef nonnull align 8 dereferenceable(129) %23)
  %25 = getelementptr inbounds i8, ptr %0, i64 72
  %26 = call noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSERKS1_(ptr noundef nonnull align 8 dereferenceable(28) %25, ptr noundef nonnull align 8 dereferenceable(28) %24)
  br label %42

27:                                               ; preds = %5
  %28 = getelementptr inbounds i8, ptr %0, i64 16
  %29 = load i8, ptr %28, align 8, !tbaa !29, !range !27, !noundef !28
  %30 = trunc nuw i8 %29 to i1
  br i1 %30, label %31, label %37

31:                                               ; preds = %27
  call void @llvm.lifetime.start.p0(i64 32, ptr nonnull %7) #6
  call void @llvm.lifetime.start.p0(i64 8, ptr nonnull %8) #6
  %32 = getelementptr inbounds i8, ptr %0, i64 18
  %33 = tail call noundef zeroext i8 @_ZNK4dali12TensorLayout4sizeEv(ptr noundef nonnull align 1 dereferenceable(16) %32) #6
  %34 = zext i8 %33 to i64
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %8, i64 noundef %34)
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind nonnull writable sret(%"struct.dali::TensorListShape") align 8 %7, i32 noundef %3, ptr noundef nonnull align 8 dereferenceable(8) %8)
  %35 = getelementptr inbounds i8, ptr %0, i64 72
  %36 = call noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %35, ptr noundef nonnull align 8 dereferenceable(28) %7)
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %7) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr nonnull %8) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr nonnull %7) #6
  br label %42

37:                                               ; preds = %27
  call void @llvm.lifetime.start.p0(i64 32, ptr nonnull %9) #6
  call void @llvm.lifetime.start.p0(i64 8, ptr nonnull %10) #6
  %38 = getelementptr inbounds i8, ptr %0, i64 40
  %39 = tail call noundef i64 @_ZNK4dali11SmallVectorIiLm6ESaIiEE4sizeEv(ptr noundef nonnull align 8 dereferenceable(32) %38)
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %10, i64 noundef %39)
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind nonnull writable sret(%"struct.dali::TensorListShape") align 8 %9, i32 noundef %3, ptr noundef nonnull align 8 dereferenceable(8) %10)
  %40 = getelementptr inbounds i8, ptr %0, i64 72
  %41 = call noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %40, ptr noundef nonnull align 8 dereferenceable(28) %9)
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %9) #6
  call void @llvm.lifetime.end.p0(i64 8, ptr nonnull %10) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr nonnull %9) #6
  br label %42

42:                                               ; preds = %31, %37, %21
  %43 = load i32, ptr %0, align 8, !tbaa !30
  %44 = and i32 %43, 4
  %.not = icmp eq i32 %44, 0
  br i1 %.not, label %62, label %45

45:                                               ; preds = %42
  call void @llvm.lifetime.start.p0(i64 8, ptr nonnull %11) #6
  %46 = sext i32 %4 to i64
  call void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8) %11, i64 noundef %46)
  %47 = getelementptr inbounds i8, ptr %0, i64 72
  %48 = call noundef i64 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE12num_elementsEv(ptr noundef nonnull align 8 dereferenceable(28) %47)
  %49 = icmp eq i64 %48, 0
  br i1 %49, label %52, label %.preheader

.preheader:                                       ; preds = %45
  %50 = call noundef i32 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE4sizeEv(ptr noundef nonnull align 8 dereferenceable(28) %47)
  %51 = icmp sgt i32 %50, 0
  br i1 %51, label %.lr.ph, label %.loopexit

52:                                               ; preds = %45
  call void @llvm.lifetime.start.p0(i64 32, ptr nonnull %12) #6
  call void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind nonnull writable sret(%"struct.dali::TensorListShape") align 8 %12, i32 noundef %3, ptr noundef nonnull align 8 dereferenceable(8) %11)
  %53 = call noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28) %47, ptr noundef nonnull align 8 dereferenceable(28) %12)
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EED2Ev(ptr noundef nonnull align 8 dereferenceable(28) %12) #6
  call void @llvm.lifetime.end.p0(i64 32, ptr nonnull %12) #6
  br label %.loopexit

.lr.ph:                                           ; preds = %.preheader, %58
  %indvars.iv = phi i64 [ %indvars.iv.next, %58 ], [ 0, %.preheader ]
  call void @llvm.lifetime.start.p0(i64 8, ptr nonnull %13) #6
  %54 = call ptr @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE17tensor_shape_spanEl(ptr noundef nonnull align 8 dereferenceable(28) %47, i64 noundef %indvars.iv)
  store ptr %54, ptr %13, align 8
  %55 = call noundef i64 @_ZN4dali6volumeINS_4spanIlLl1EEEEEDTcl6volumeclsr4daliE5beginfp_Eclsr4daliE3endfp_EEERKT_(ptr noundef nonnull align 8 dereferenceable(8) %13)
  %56 = icmp eq i64 %55, 0
  call void @llvm.lifetime.end.p0(i64 8, ptr nonnull %13) #6
  br i1 %56, label %57, label %58

57:                                               ; preds = %.lr.ph
  call void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE16set_tensor_shapeElRKNS_11TensorShapeILi1EEE(ptr noundef nonnull align 8 dereferenceable(28) %47, i64 noundef %indvars.iv, ptr noundef nonnull align 8 dereferenceable(8) %11)
  br label %58

58:                                               ; preds = %.lr.ph, %57
  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  %59 = call noundef i32 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE4sizeEv(ptr noundef nonnull align 8 dereferenceable(28) %47)
  %60 = sext i32 %59 to i64
  %61 = icmp slt i64 %indvars.iv.next, %60
  br i1 %61, label %.lr.ph, label %.loopexit, !llvm.loop !31

.loopexit:                                        ; preds = %58, %.preheader, %52
  call void @llvm.lifetime.end.p0(i64 8, ptr nonnull %11) #6
  br label %62

62:                                               ; preds = %.loopexit, %42
  ret void
}

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef zeroext i1 @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEcvbEv(ptr noundef nonnull align 8 dereferenceable(8)) local_unnamed_addr #3 align 2

; Function Attrs: noreturn nounwind
declare void @__assert_fail(ptr noundef, ptr noundef, i32 noundef, ptr noundef) local_unnamed_addr #4

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef ptr @_ZNKSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEptEv(ptr noundef nonnull align 8 dereferenceable(8)) local_unnamed_addr #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali8ArgValueIiLi1EE7AcquireINS_16ArgShapeFromSizeILi1EEEEEvRKNS_6OpSpecERKNS_17ArgumentWorkspaceEiNS_13ArgValueFlagsEOT_(ptr noundef nonnull align 8 dereferenceable(129), ptr noundef nonnull align 8 dereferenceable(328), ptr noundef nonnull align 8 dereferenceable(256), i32 noundef, i32 noundef, ptr noundef nonnull align 1 dereferenceable(1)) local_unnamed_addr #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef nonnull align 8 dereferenceable(56) ptr @_ZNK4dali8ArgValueIiLi1EE3getEv(ptr noundef nonnull align 8 dereferenceable(129)) local_unnamed_addr #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSERKS1_(ptr noundef nonnull align 8 dereferenceable(28), ptr noundef nonnull align 8 dereferenceable(28)) local_unnamed_addr #2 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali18uniform_list_shapeILin2ENS_11TensorShapeILi1EEELi1ELi1EEENS_15TensorListShapeIXT2_EEEiRKT0_(ptr dead_on_unwind noalias writable sret(%"struct.dali::TensorListShape") align 8, i32 noundef, ptr noundef nonnull align 8 dereferenceable(8)) local_unnamed_addr #2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef zeroext i8 @_ZNK4dali12TensorLayout4sizeEv(ptr noundef nonnull align 1 dereferenceable(16)) local_unnamed_addr #3 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali11TensorShapeILi1EEC2IJEEElDpT_(ptr noundef nonnull align 8 dereferenceable(8), i64 noundef) unnamed_addr #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef nonnull align 8 dereferenceable(28) ptr @_ZN4dali15TensorListShapeILi1EEaSEOS1_(ptr noundef nonnull align 8 dereferenceable(28), ptr noundef nonnull align 8 dereferenceable(28)) local_unnamed_addr #3 align 2

; Function Attrs: inlinehint mustprogress nounwind uwtable
declare dso_local noundef i64 @_ZNK4dali11SmallVectorIiLm6ESaIiEE4sizeEv(ptr noundef nonnull align 8 dereferenceable(32)) local_unnamed_addr #1 align 2

; Function Attrs: mustprogress uwtable
declare dso_local noundef i64 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE12num_elementsEv(ptr noundef nonnull align 8 dereferenceable(28)) local_unnamed_addr #2 align 2

; Function Attrs: mustprogress nounwind uwtable
declare dso_local noundef i32 @_ZNK4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE4sizeEv(ptr noundef nonnull align 8 dereferenceable(28)) local_unnamed_addr #3 align 2

; Function Attrs: inlinehint mustprogress uwtable
declare dso_local noundef i64 @_ZN4dali6volumeINS_4spanIlLl1EEEEEDTcl6volumeclsr4daliE5beginfp_Eclsr4daliE3endfp_EEERKT_(ptr noundef nonnull align 8 dereferenceable(8)) local_unnamed_addr #5

; Function Attrs: mustprogress uwtable
declare dso_local ptr @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE17tensor_shape_spanEl(ptr noundef nonnull align 8 dereferenceable(28), i64 noundef) local_unnamed_addr #2 align 2

; Function Attrs: mustprogress uwtable
declare dso_local void @_ZN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EE16set_tensor_shapeElRKNS_11TensorShapeILi1EEE(ptr noundef nonnull align 8 dereferenceable(28), i64 noundef, ptr noundef nonnull align 8 dereferenceable(8)) local_unnamed_addr #2 align 2

attributes #0 = { mustprogress nocallback nofree nosync nounwind willreturn memory(argmem: readwrite) }
attributes #1 = { inlinehint mustprogress nounwind uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #2 = { mustprogress uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #3 = { mustprogress nounwind uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #4 = { noreturn nounwind "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #5 = { inlinehint mustprogress uwtable "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cmov,+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #6 = { nounwind }
attributes #7 = { noreturn nounwind }

!llvm.module.flags = !{!0, !1, !2, !3}
!llvm.ident = !{!4}

!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{i32 8, !"PIC Level", i32 2}
!2 = !{i32 7, !"PIE Level", i32 2}
!3 = !{i32 7, !"uwtable", i32 2}
!4 = !{!"clang version 19.1.0 (git@github.com:llvm/llvm-project.git a4bf6cd7cfb1a1421ba92bca9d017b49936c55e4)"}
!5 = !{!6, !17, i64 17}
!6 = !{!"_ZTSN4dali8AxisArgsE", !7, i64 0, !10, i64 8, !17, i64 16, !17, i64 17, !18, i64 18, !19, i64 40, !21, i64 72}
!7 = !{!"int", !8, i64 0}
!8 = !{!"omnipotent char", !9, i64 0}
!9 = !{!"Simple C++ TBAA"}
!10 = !{!"_ZTSSt10unique_ptrIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EE", !11, i64 0}
!11 = !{!"_ZTSSt15__uniq_ptr_dataIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_ELb1ELb1EE", !12, i64 0}
!12 = !{!"_ZTSSt15__uniq_ptr_implIN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EE", !13, i64 0}
!13 = !{!"_ZTSSt5tupleIJPN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEE", !14, i64 0}
!14 = !{!"_ZTSSt11_Tuple_implILm0EJPN4dali8ArgValueIiLi1EEESt14default_deleteIS2_EEE", !15, i64 0}
!15 = !{!"_ZTSSt10_Head_baseILm0EPN4dali8ArgValueIiLi1EEELb0EE", !16, i64 0}
!16 = !{!"any pointer", !8, i64 0}
!17 = !{!"bool", !8, i64 0}
!18 = !{!"_ZTSN4dali12TensorLayoutE", !8, i64 0}
!19 = !{!"_ZTSN4dali11SmallVectorIiLm6ESaIiEEE", !8, i64 0, !20, i64 24}
!20 = !{!"long", !8, i64 0}
!21 = !{!"_ZTSN4dali15TensorListShapeILi1EEE", !22, i64 0}
!22 = !{!"_ZTSN4dali19TensorListShapeBaseINS_15TensorListShapeILi1EEELi1EEE", !23, i64 0, !7, i64 24}
!23 = !{!"_ZTSSt6vectorIlSaIlEE", !24, i64 0}
!24 = !{!"_ZTSSt12_Vector_baseIlSaIlEE", !25, i64 0}
!25 = !{!"_ZTSNSt12_Vector_baseIlSaIlEE12_Vector_implE", !26, i64 0}
!26 = !{!"_ZTSNSt12_Vector_baseIlSaIlEE17_Vector_impl_dataE", !16, i64 0, !16, i64 8, !16, i64 16}
!27 = !{i8 0, i8 2}
!28 = !{}
!29 = !{!6, !17, i64 16}
!30 = !{!6, !7, i64 0}
!31 = distinct !{!31, !32}
!32 = !{!"llvm.loop.mustprogress"}


; Effective O3 passes: EarlyCSEPass,GlobalOptPass,IPSCCPPass,IndVarSimplifyPass,InferFunctionAttrsPass,InstCombinePass,JumpThreadingPass,LoopRotatePass,LoopSimplifyPass,LoopUnrollPass,SROAPass,SimplifyCFGPass,TailCallElimPass