Skip to content

Commit

Permalink
remove debug script
Browse files Browse the repository at this point in the history
  • Loading branch information
wxj6000 committed Dec 9, 2024
1 parent df7a1ff commit 7dca6ae
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 18 deletions.
14 changes: 0 additions & 14 deletions gpu4pyscf/lib/gvhf-md/unrolled_md_j.cu
Original file line number Diff line number Diff line change
Expand Up @@ -5018,20 +5018,6 @@ int md_j_unrolled(RysIntEnvVars *envs, JKMatrix *jk, BoundsInfo *bounds,
dim3 threads(16, 16);
dim3 blocks;
int ijkl = lij*9 + lkl;
printf("%d %d %d %d %d\n", ijkl, li, lj, lk, ll);
cudaFuncAttributes funcAttributes;
// Query the max dynamic shared memory size attribute
cudaError_t err = cudaFuncGetAttributes(
&funcAttributes,
md_j_2_0
);

if (err != cudaSuccess) {
printf("Error querying function attribute: %s\n", cudaGetErrorString(err));
return -1;
}

printf("Max Dynamic Shared Memory Size: %d bytes\n", funcAttributes.cudaFuncAttributeMaxDynamicSharedMemorySize);

switch (ijkl) {
case 0: // lij=0, lkl=0
Expand Down
5 changes: 1 addition & 4 deletions gpu4pyscf/scf/j_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,17 +96,14 @@ def get_j(mol, dm, hermi=1, vhfopt=None, omega=None, verbose=None):
pair_loc_on_gpu.data.ptr,
)

libvhf_md.init_mdj_constant(ctypes.c_int(SHM_SIZE))

uniq_l_ctr = vhfopt.uniq_l_ctr
uniq_l = uniq_l_ctr[:,0]
l_ctr_bas_loc = vhfopt.l_ctr_offsets
l_symb = [lib.param.ANGULAR[i] for i in uniq_l]
n_groups = len(uniq_l_ctr)
tile_mappings = {}
workers = gpu_specs['multiProcessorCount']
print(gpu_specs)
exit()

info = cp.empty(2, dtype=np.uint32)

for i in range(n_groups):
Expand Down

0 comments on commit 7dca6ae

Please sign in to comment.