48 lines
996 B
Plaintext
48 lines
996 B
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"# prerequisites: KrylovKit, TensorOperations, LinearAlgebra, CUDA#tb/cutensor (for GPU mode)\n",
|
|
"\n",
|
|
"include(\"CPU.jl\") # using CPU mode\n",
|
|
"T = Float32\n",
|
|
"\n",
|
|
"V_test(r2::T)::T =\n",
|
|
" -4 * exp(-r2 / 4)\n",
|
|
"\n",
|
|
"d = 3\n",
|
|
"n = 3\n",
|
|
"N = 6\n",
|
|
"L::T = 12.0\n",
|
|
"mu::T = 0.5\n",
|
|
"n_imag = 1\n",
|
|
"\n",
|
|
"H = HOperator{T}(V_test, 3, 3, N, L, 0.5f0, 1)\n",
|
|
"@time evals, evecs, info = eig(H, 5)\n",
|
|
"print(info.numops, \" operations : \")\n",
|
|
"println(evals)"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Julia 1.8.5",
|
|
"language": "julia",
|
|
"name": "julia-1.8"
|
|
},
|
|
"language_info": {
|
|
"file_extension": ".jl",
|
|
"mimetype": "application/julia",
|
|
"name": "julia",
|
|
"version": "1.8.5"
|
|
},
|
|
"orig_nbformat": 4
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 2
|
|
}
|