yapanuwan
/
DVR-jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' into debugging

This commit is contained in:
ysyapa 2023-04-03 20:56:51 -04:00
parent 94acb12d5f 526ef37578
commit 3daddf7a9a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
example.ipynb
View File

@ -6,22 +6,22 @@
"metadata": {},
"outputs": [],
"source": [
"# prerequisites: KrylovKit, TensorOperations, LinearAlgebra, CUDA#tb/cutensor (for GPU mode)\n",
"# prerequisite packages: KrylovKit, TensorOperations, LinearAlgebra, CUDA#tb/cutensor\n",
"\n",
"include(\"CPU.jl\") # using CPU mode\n",
"T = Float32\n",
"\n",
"V_test(r2::T)::T =\n",
"V_gauss(r2::T)::T =\n",
" -4 * exp(-r2 / 4)\n",
"\n",
"d = 3\n",
"n = 3\n",
"N = 6\n",
"L::T = 12.0\n",
"L::T = 12\n",
"mu::T = 0.5\n",
"n_imag = 1\n",
"\n",
"H = HOperator{T}(V_test, 3, 3, N, L, 0.5f0, 1)\n",
"H = HOperator{T}(V_gauss, d, n, N, L, mu, n_imag)\n",
"@time evals, evecs, info = eig(H, 5)\n",
"print(info.numops, \" operations : \")\n",
"println(evals)"