Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add speed notebook #1635

Closed
wants to merge 2 commits into from
Closed

Add speed notebook #1635

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8ff3b91
🙈 Ignore .ipynb_checkpoints
blegat Nov 19, 2018
c4369d4
⚡️ Add speed notebook
blegat Nov 19, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions test/.gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
.ipynb_checkpoints
272 changes: 272 additions & 0 deletions test/perf/speed.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,272 @@
{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"using Random\n",
"using JuMP\n",
"using TimerOutputs"
]
},
{
"cell_type": "code",
"execution_count": 30,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"p_median (generic function with 1 method)"
]
},
"execution_count": 30,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"function p_median(to, num_facilities, num_customers, num_locations)\n",
" @timeit to \"constants\" begin\n",
" Random.seed!(10)\n",
" customer_locations = [rand(1:num_locations) for _ in 1:num_customers]\n",
" end\n",
"\n",
" @timeit to \"model\" begin\n",
" model = Model()\n",
" end\n",
" @timeit to \"variable\" begin\n",
" @timeit to \"facility\" begin\n",
" has_facility = @variable(model, [1:num_locations], Bin)\n",
" end\n",
" @timeit to \"closest\" begin\n",
" is_closest = @variable(model, [1:num_locations, 1:num_customers], Bin)\n",
" end\n",
" end\n",
"\n",
" @timeit to \"objective\" begin\n",
" @objective(model, Min,\n",
" sum(abs(customer_locations[customer] - location)\n",
" * is_closest[location, customer]\n",
" for customer in 1:num_customers, location in 1:num_locations))\n",
" end\n",
"\n",
" @timeit to \"constraint\" begin\n",
" @timeit to \"customer loop\" begin\n",
" for customer in 1:num_customers\n",
" @timeit to \"facility\" begin\n",
" # `location` can't be closest for `customer` if there is no facility.\n",
" @constraint(model,\n",
" [location in 1:num_locations],\n",
" is_closest[location, customer] <= has_facility[location])\n",
" end\n",
" @timeit to \"closest\" begin\n",
" # One facility must be the closest for `customer`.\n",
" @constraint(model,\n",
" sum(is_closest[location, customer]\n",
" for location in 1:num_locations) == 1)\n",
" end\n",
" end\n",
" end\n",
"\n",
" @timeit to \"sum facility\" begin\n",
" # Must place all facilities.\n",
" @constraint(model, sum(has_facility) == num_facilities)\n",
" end\n",
" end\n",
" \n",
" return\n",
"end"
]
},
{
"cell_type": "code",
"execution_count": 38,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
" \u001b[1m────────────────────────────────────────────────────────────────────────────\u001b[22m\n",
" \u001b[1m \u001b[22m Time Allocations \n",
" ────────────────────── ───────────────────────\n",
" Tot / % measured: 2.22s / 99.5% 1.01GiB / 100% \n",
"\n",
" Section ncalls time %tot avg alloc %tot avg\n",
" ────────────────────────────────────────────────────────────────────────────\n",
" P median 1 2.21s 100% 2.21s 1.01GiB 100% 1.01GiB\n",
" constraint 1 1.61s 73.0% 1.61s 898MiB 86.8% 898MiB\n",
" customer loop 1 1.61s 72.9% 1.61s 897MiB 86.7% 897MiB\n",
" facility 100 1.44s 65.4% 14.4ms 848MiB 82.0% 8.48MiB\n",
" closest 100 166ms 7.53% 1.66ms 48.9MiB 4.73% 501KiB\n",
" sum facility 1 658μs 0.03% 658μs 870KiB 0.08% 870KiB\n",
" variable 1 402ms 18.2% 402ms 104MiB 10.1% 104MiB\n",
" closest 1 401ms 18.2% 401ms 103MiB 9.93% 103MiB\n",
" facility 1 1.12ms 0.05% 1.12ms 1.34MiB 0.13% 1.34MiB\n",
" objective 1 194ms 8.81% 194ms 32.6MiB 3.15% 32.6MiB\n",
" model 1 25.8μs 0.00% 25.8μs 22.3KiB 0.00% 22.3KiB\n",
" constants 1 20.8μs 0.00% 20.8μs 0.98KiB 0.00% 0.98KiB\n",
" \u001b[1m────────────────────────────────────────────────────────────────────────────\u001b[22m"
]
},
"execution_count": 38,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"to = TimerOutput()\n",
"@timeit to \"P median\" begin\n",
" p_median(to, 100, 100, 5000);\n",
"end\n",
"to"
]
},
{
"cell_type": "code",
"execution_count": 32,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"cont5 (generic function with 2 methods)"
]
},
"execution_count": 32,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"function cont5(to, n)\n",
" @timeit to \"constants\" begin\n",
" m = n\n",
" n1 = n-1\n",
" m1 = m-1\n",
" dx = 1/n\n",
" T = 1.58\n",
" dt = T/m\n",
" h2 = dx^2\n",
" a = 0.001\n",
" yt = [0.5*(1 - (j*dx)^2) for j in 0:n]\n",
" end\n",
"\n",
" @timeit to \"model\" begin\n",
" model = Model()\n",
" end\n",
" @timeit to \"variable\" begin\n",
" @timeit to \"y\" begin\n",
" y = @variable(model, [0:m,0:n], lower_bound=0, upper_bound=1)\n",
" end\n",
" @timeit to \"u\" begin\n",
" u = @variable(model, [1:m], lower_bound=-1, upper_bound=1)\n",
" end\n",
" end\n",
"\n",
" @timeit to \"objective\" begin\n",
" @objective(model, Min, y[0,0])\n",
" end\n",
"\n",
" @timeit to \"constraint\" begin\n",
" @timeit to \"PDE\" begin\n",
" for i in 0:m1\n",
" for j in 1:n1\n",
" @constraint(model,\n",
" h2*(y[i+1,j] - y[i,j])\n",
" == 0.5*dt*(y[i,j-1] - 2*y[i,j] +\n",
" y[i,j+1] + y[i+1,j-1] - 2*y[i+1,j] + y[i+1,j+1]) )\n",
" end\n",
" end\n",
" end\n",
"\n",
" @timeit to \"Initial conditions\" begin\n",
" for j in 0:n\n",
" @constraint(model, y[0,j] == 0)\n",
" end\n",
" end\n",
"\n",
" @timeit to \"Boundary conditions\" begin\n",
" for i in 1:m\n",
" @constraint(model,\n",
" y[i,2] - 4*y[i,1] + 3*y[i,0] == 0)\n",
" @constraint(model,\n",
" y[i,n-2] - 4*y[i,n1] + 3*y[i,n]== (2*dx)*(u[i] - y[i,n]))\n",
" end\n",
" end\n",
" end\n",
" \n",
" return\n",
"end"
]
},
{
"cell_type": "code",
"execution_count": 35,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
" \u001b[1m──────────────────────────────────────────────────────────────────────────────\u001b[22m\n",
" \u001b[1m \u001b[22m Time Allocations \n",
" ────────────────────── ───────────────────────\n",
" Tot / % measured: 1.18s / 98.7% 452MiB / 100% \n",
"\n",
" Section ncalls time %tot avg alloc %tot avg\n",
" ──────────────────────────────────────────────────────────────────────────────\n",
" cont5 1 1.17s 100% 1.17s 452MiB 100% 452MiB\n",
" constraint 1 923ms 79.1% 923ms 360MiB 79.6% 360MiB\n",
" PDE 1 921ms 78.9% 921ms 357MiB 79.0% 357MiB\n",
" Boundary condi... 1 1.88ms 0.16% 1.88ms 2.04MiB 0.45% 2.04MiB\n",
" Initial condit... 1 459μs 0.04% 459μs 540KiB 0.12% 540KiB\n",
" variable 1 244ms 20.9% 244ms 92.3MiB 20.4% 92.3MiB\n",
" y 1 244ms 20.9% 244ms 92.3MiB 20.4% 92.3MiB\n",
" u 1 284μs 0.02% 284μs 74.4KiB 0.02% 74.4KiB\n",
" constants 1 54.9μs 0.00% 54.9μs 35.5KiB 0.01% 35.5KiB\n",
" model 1 25.2μs 0.00% 25.2μs 22.3KiB 0.00% 22.3KiB\n",
" objective 1 3.07μs 0.00% 3.07μs - 0.00% -\n",
" \u001b[1m──────────────────────────────────────────────────────────────────────────────\u001b[22m"
]
},
"execution_count": 35,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"to = TimerOutput()\n",
"@timeit to \"cont5\" begin\n",
" cont5(to, 500);\n",
"end\n",
"to"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Julia 1.0.3-pre",
"language": "julia",
"name": "julia-1.0"
},
"language_info": {
"file_extension": ".jl",
"mimetype": "application/julia",
"name": "julia",
"version": "1.0.3"
}
},
"nbformat": 4,
"nbformat_minor": 2
}