wg< zddlmZddlmZddlmZmZmZdZGddeZ e dk(rejyy) )cuda) TypingError)unittest CUDATestCaseskip_on_cudasimcyNxs h/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_errors.pynooprscJeZdZdZdZdZdZdZdZe ddZ y ) TestJitErrorsz- Test compile-time errors with @jit. ctjt}|jt5}|dddd|j dt j|jt5}|dddd|j dt |jy#1swYwxYw#1swY;xYw)N)))zBgriddim must be a sequence of 1, 2 or 3 integers, got [1, 2, 3, 4]))rr)rrrrzCblockdim must be a sequence of 1, 2 or 3 integers, got [3, 4, 5, 6])rjitr assertRaises ValueErrorassertInstr exceptionselfkernfuncraisess r test_too_many_dimsz TestJitErrors.test_too_many_dimss88D>   z * +f ) * + )&**+ -  z * ,f * + , )&**+ - + +  , ,B+7B7+B47Cctjt}|jt5}|dddd|j dt j|jt5}|dddd|j dt |jy#1swYwxYw#1swY;xYw)N)g@rz1griddim must be a sequence of integers, got [2.0])rg@z2blockdim must be a sequence of integers, got [3.0])rrrr TypeErrorrrrrs r test_non_integral_dimsz$TestJitErrors.test_non_integral_dimss88D>   y ) V V   I&**+ -  y ) V V   J&**+ -     r$c|jt5}|dddd|jdtjy#1swY/xYw)Nrz&launch configuration was not specified)rrrrrrs r _test_unconfiguredz TestJitErrors._test_unconfigured+sK   z * f QK  >&**+ -  s A  Acdtjdt}|j|y)Nz void(int32)rrrr)r r!s r "test_unconfigured_typed_cudakernelz0TestJitErrors.test_unconfigured_typed_cudakernel1s%*488M*40 )rcXtjt}|j|yr r+r,s r $test_unconfigured_untyped_cudakernelz2TestJitErrors.test_unconfigured_untyped_cudakernel5s88D> )rz'TypingError does not occur on simulatorcHtjddtjfd}|jt5}|ddddt j }|j d||j d|y#1swYCxYw)NT)devicect|Sr )floorr s r dev_funcz1TestJitErrors.test_typing_error..dev_func>s 8Orcdy)Ng?r )r4sr kernel_funcz4TestJitErrors.test_typing_error..kernel_funcCs SMr)rrz*resolving callee type: type(CUDADispatcherz&NameError: name 'floor' is not defined)rrrrrrr)r r6r"excstrr4s @r test_typing_errorzTestJitErrors.test_typing_error9s           { + v K   V%%& BFK >G   s BB!N) __name__ __module__ __qualname____doc__r#r'r)r-r/rr8r rr rr s= - -- **>?H@Hrr__main__N) numbarnumba.core.errorsrnumba.cuda.testingrrrrrr9mainr rr rBsA)FF AHLAHH zHMMOr