diff --git a/example.ipynb b/example.ipynb index dc235e1..ad9cd64 100644 --- a/example.ipynb +++ b/example.ipynb @@ -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)"