Module ImportFrom alias alias alias Import alias identifier Import alias ImportFrom alias alias alias alias ImportFrom alias If AttributeLoad NameLoad attr body Assign NameStore Num orelse Assign NameStore Num Assign NameStore TupleLoad NameLoad NameLoad ClassDef bases AttributeLoad NameLoad attr body FunctionDef arguments args NameParam defaults body FunctionDef arguments args NameParam NameParam defaults body If CompareGt NameLoad NameLoad body Return NameLoad orelse Return NameLoad decorator_list Call AttributeLoad NameLoad attr Call NameLoad NameLoad NameLoad keyword NameLoad keyword NameLoad FunctionDef arguments args NameParam NameParam NameParam defaults body Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad keyword NameLoad Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore SubscriptLoad AttributeLoad NameLoad attr Index Num Assign NameStore SubscriptLoad AttributeLoad NameLoad attr Index Num Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr Num Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad Num If BoolOpAnd CompareGtE NameLoad Num CompareLt NameLoad BinOpSub NameLoad Num CompareGtE NameLoad Num CompareLt NameLoad BinOpSub NameLoad Num body Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad BinOpMult Num BinOpAdd BinOpAdd BinOpAdd SubscriptLoad NameLoad Index TupleLoad NameLoad BinOpAdd NameLoad Num SubscriptLoad NameLoad Index TupleLoad NameLoad BinOpSub NameLoad Num SubscriptLoad NameLoad Index TupleLoad BinOpSub NameLoad Num NameLoad SubscriptLoad NameLoad Index TupleLoad BinOpAdd NameLoad Num NameLoad Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad BinOpSub SubscriptLoad NameLoad Index TupleLoad NameLoad NameLoad SubscriptLoad NameLoad Index TupleLoad NameLoad NameLoad Expr Call AttributeLoad NameLoad attr Assign NameStore BinOpFloorDiv NameLoad Num While CompareGt NameLoad Num body If CompareLt NameLoad NameLoad body Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad Call NameLoad SubscriptLoad NameLoad Index TupleLoad NameLoad NameLoad SubscriptLoad NameLoad Index TupleLoad BinOpAdd NameLoad NameLoad NameLoad AugAssignFloorDiv NameStore Num Expr Call AttributeLoad NameLoad attr Assign NameStore BinOpFloorDiv NameLoad Num While CompareGt NameLoad Num body If BoolOpAnd CompareLt NameLoad NameLoad CompareEq NameLoad Num body Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad Call NameLoad SubscriptLoad NameLoad Index TupleLoad NameLoad NameLoad SubscriptLoad NameLoad Index TupleLoad NameLoad BinOpAdd NameLoad NameLoad AugAssignFloorDiv NameStore Num Expr Call AttributeLoad NameLoad attr If BoolOpAnd CompareEq NameLoad Num CompareEq NameLoad Num body Assign SubscriptStore NameLoad Index TupleLoad NameLoad NameLoad SubscriptLoad NameLoad Index TupleLoad Num Num decorator_list Call AttributeLoad NameLoad attr Call NameLoad SubscriptLoad NameLoad ExtSlice Slice Slice SubscriptLoad NameLoad ExtSlice Slice Slice SubscriptLoad NameLoad ExtSlice Slice Slice If AttributeLoad NameLoad attr body Assign TupleStore NameStore NameStore TupleLoad Num Num Assign NameStore Num orelse Assign TupleStore NameStore NameStore TupleLoad Num Num Assign NameStore Num Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad NameLoad keyword AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad NameLoad keyword AttributeLoad NameLoad attr Assign NameStore NameLoad Assign NameStore NameLoad Assign NameStore Num Assign NameStore Num For NameStore Call NameLoad NameLoad body Assign SubscriptStore NameLoad Index TupleLoad NameLoad Num Num Assign SubscriptStore NameLoad Index TupleLoad NameLoad Num Num Assign NameStore Call AttributeLoad NameLoad attr Assign NameStore Num Assign NameStore TupleLoad NameLoad NameLoad Assign NameStore TupleLoad BinOpFloorDiv NameLoad SubscriptLoad NameLoad Index Num BinOpFloorDiv NameLoad SubscriptLoad NameLoad Index Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad While BoolOpAnd CompareGt NameLoad NameLoad CompareLt NameLoad NameLoad body Expr Call AttributeLoad NameLoad attr CompareEq AttributeLoad NameLoad attr AttributeLoad NameLoad attr Expr Call SubscriptLoad NameLoad Index TupleLoad NameLoad NameLoad NameLoad NameLoad NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad keyword NameLoad Expr Call AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr NameLoad attr Assign NameStore NameLoad Assign NameStore NameLoad Assign NameStore NameLoad AugAssignAdd NameStore Num Assign NameStore BinOpSub Call AttributeLoad NameLoad attr NameLoad decorator_list decorator_list If CompareEq NameLoad Str body Expr Call AttributeLoad NameLoad attr
N/A __future__ print_function absolute_import division N/A numpy np N/A time numba cuda config float64 void numba.cuda.testing unittest N/A N/A config ENABLE_CUDASIM N/A N/A tpb 4 N/A N/A tpb 16 N/A SM_SIZE N/A tpb tpb TestCudaLaplace N/A N/A unittest TestCase N/A test_laplace_small N/A N/A self N/A N/A get_max N/A N/A a b N/A N/A N/A N/A a b N/A N/A a N/A N/A b N/A N/A N/A cuda jit N/A float64 float64 float64 device True inline True jocabi_relax_core N/A N/A A Anew error N/A N/A N/A err_sm N/A N/A N/A cuda shared array SM_SIZE dtype float64 N/A ty N/A N/A cuda threadIdx x N/A tx N/A N/A cuda threadIdx y N/A bx N/A N/A cuda blockIdx x N/A by N/A N/A cuda blockIdx y N/A n N/A N/A A shape N/A 0 N/A m N/A N/A A shape N/A 1 N/A N/A i j N/A N/A cuda grid 2 N/A N/A err_sm N/A N/A ty tx 0 N/A N/A N/A j 1 N/A j N/A n 1 N/A i 1 N/A i N/A m 1 N/A N/A N/A Anew N/A N/A j i N/A 0.25 N/A N/A N/A N/A A N/A N/A j N/A i 1 N/A A N/A N/A j N/A i 1 N/A A N/A N/A N/A j 1 i N/A A N/A N/A N/A j 1 i N/A N/A err_sm N/A N/A ty tx N/A N/A Anew N/A N/A j i N/A A N/A N/A j i N/A N/A N/A cuda syncthreads N/A t N/A tpb 2 N/A N/A t 0 N/A N/A N/A ty t N/A N/A N/A err_sm N/A N/A ty tx N/A get_max N/A err_sm N/A N/A ty tx N/A err_sm N/A N/A N/A ty t tx N/A t 2 N/A N/A N/A cuda syncthreads N/A t N/A tpb 2 N/A N/A t 0 N/A N/A N/A N/A tx t N/A ty 0 N/A N/A N/A err_sm N/A N/A ty tx N/A get_max N/A err_sm N/A N/A ty tx N/A err_sm N/A N/A ty N/A tx t N/A t 2 N/A N/A N/A cuda syncthreads N/A N/A N/A tx 0 N/A ty 0 N/A N/A N/A error N/A N/A by bx N/A err_sm N/A N/A 0 0 N/A N/A N/A cuda jit N/A void N/A float64 N/A N/A N/A N/A float64 N/A N/A N/A N/A float64 N/A N/A N/A N/A N/A config ENABLE_CUDASIM N/A N/A N/A NN NM N/A 4 4 N/A iter_max 20 N/A N/A N/A NN NM N/A 256 256 N/A iter_max 1000 N/A A N/A N/A np zeros N/A NN NM dtype N/A np float64 N/A Anew N/A N/A np zeros N/A NN NM dtype N/A np float64 N/A n NN N/A m NM N/A tol 1e-06 N/A error 1.0 N/A j N/A range n N/A N/A N/A A N/A N/A j 0 1.0 N/A N/A Anew N/A N/A j 0 1.0 N/A timer N/A N/A time time N/A iter 0 N/A blockdim N/A tpb tpb N/A griddim N/A N/A NN N/A blockdim N/A 0 N/A NM N/A blockdim N/A 1 N/A error_grid N/A N/A np zeros griddim N/A stream N/A N/A cuda stream N/A dA N/A N/A cuda to_device A stream N/A dAnew N/A N/A cuda to_device Anew stream N/A derror_grid N/A N/A cuda to_device error_grid stream N/A N/A N/A error tol N/A iter iter_max N/A N/A N/A N/A self assertTrue N/A N/A error_grid dtype N/A np float64 N/A N/A N/A jocabi_relax_core N/A N/A griddim blockdim stream dA dAnew derror_grid N/A N/A N/A derror_grid copy_to_host error_grid stream stream N/A N/A N/A stream synchronize N/A error N/A N/A N/A N/A np abs error_grid max N/A tmp dA N/A dA dAnew N/A dAnew tmp N/A iter 1 N/A runtime N/A N/A N/A time time timer N/A N/A N/A N/A __name__ __main__ N/A N/A N/A N/A unittest main