Skip to content

Commit

Permalink
Minor.
Browse files Browse the repository at this point in the history
  • Loading branch information
hhoppe committed Dec 3, 2024
1 parent c5e3952 commit c75fc48
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 16 deletions.
35 changes: 19 additions & 16 deletions resampler_notebook.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -5445,14 +5445,14 @@
},
{
"cell_type": "code",
"execution_count": 98,
"execution_count": 192,
"metadata": {
"execution": {
"iopub.execute_input": "2024-11-29T16:43:52.340261Z",
"iopub.status.busy": "2024-11-29T16:43:52.339759Z",
"iopub.status.idle": "2024-11-29T16:44:00.909881Z",
"shell.execute_reply": "2024-11-29T16:44:00.909234Z",
"shell.execute_reply.started": "2024-11-29T16:43:52.340244Z"
"iopub.execute_input": "2024-12-02T21:49:40.127194Z",
"iopub.status.busy": "2024-12-02T21:49:40.126583Z",
"iopub.status.idle": "2024-12-02T21:49:48.381747Z",
"shell.execute_reply": "2024-12-02T21:49:48.380773Z",
"shell.execute_reply.started": "2024-12-02T21:49:40.127174Z"
},
"lines_to_next_cell": 2,
"tags": []
Expand All @@ -5463,21 +5463,21 @@
"output_type": "stream",
"text": [
"# ** (2000, 4000, 3) -> (100, 200) trapezoid float32:\n",
"# resize_serial : 0.004 s\n",
"# resize_in_numpy : 0.004 s\n",
"# resize_in_tensorflow: 0.066 s\n",
"# resize_in_torch : 0.009 s\n",
"# resize_in_jax : 0.055 s\n",
"# resize_serial : 0.003 s\n",
"# resize_in_numpy : 0.003 s\n",
"# resize_in_tensorflow: 0.059 s\n",
"# resize_in_torch : 0.007 s\n",
"# resize_in_jax : 0.054 s\n",
"# jaxjit_resize : 0.024 s\n",
"# tf_image_resize : 0.062 s\n",
"# tf_image_resize : 0.063 s\n",
"# reshape_mean : 0.106 s\n",
"# reshape2 : 0.104 s\n",
"# reshape3 : 0.114 s\n",
"# reshape3 : 0.117 s\n",
"# einsum : 0.040 s\n",
"# two_dots : 0.040 s\n",
"# two_dots : 0.041 s\n",
"# reshape4 : 0.037 s\n",
"# reshape5 : 0.028 s\n",
"# reduceat : 0.018 s\n"
"# reshape5 : 0.029 s\n",
"# reduceat : 0.017 s\n"
]
}
],
Expand Down Expand Up @@ -5594,6 +5594,9 @@
" test_profile_downsampling((2000, 2000, 4), (1000, 1000))\n",
" test_profile_downsampling((5000, 5000, 1), (1000, 1000), also_prun=True)\n",
"if EFFORT >= 2:\n",
" # Occasional Numba error when invoking `jitted_function(a)`:\n",
" # RuntimeError: In 'NRT_adapt_ndarray_to_python', 'descr' is NULL.\n",
" # The solution was to delete resampler/__pycache__.\n",
" test_profile_downsampling((8192, 8192, 3), (2048, 2048))\n",
" test_profile_downsampling((8192, 8192, 1), (2048, 2048))"
]
Expand Down
3 changes: 3 additions & 0 deletions resampler_notebook.py
Original file line number Diff line number Diff line change
Expand Up @@ -2159,6 +2159,9 @@ def reduceat() -> _NDArray:
test_profile_downsampling((2000, 2000, 4), (1000, 1000))
test_profile_downsampling((5000, 5000, 1), (1000, 1000), also_prun=True)
if EFFORT >= 2:
# Occasional Numba error when invoking `jitted_function(a)`:
# RuntimeError: In 'NRT_adapt_ndarray_to_python', 'descr' is NULL.
# The solution was to delete resampler/__pycache__.
test_profile_downsampling((8192, 8192, 3), (2048, 2048))
test_profile_downsampling((8192, 8192, 1), (2048, 2048))

Expand Down

0 comments on commit c75fc48

Please sign in to comment.