fromnose.toolsimportraisesimporttvm@raises(Exception)deftest_loop_dependent_allocate():N=tvm.var("N")A=tvm.placeholder((2*N,),"float32","A")C=tvm.compute((N,),lambdai:A[2*i]+A[i+1],name='C')s=tvm.create_schedule(C.op)AA=s.cache_read(A,"local",[C])s[AA].compute_at(s[C],s[C].op.axis[0])# this line should fail due to IRUseDefAnalysis sees an allocate statement# referencing undefined variabletvm.lower(s,[A,C])if__name__=="__main__":test_loop_dependent_allocate()