{ "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 }