search

NVIDIA / TorchFort

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
https://nvidia.github.io/TorchFort/
Other
154 stars 19 forks source link

Fix interface issues in Fortran module with gfortran. #18

Closed romerojosh closed 2 months ago

romerojosh commented 2 months ago

Fixing some compilation issues of the Torchfort module with gfortran:

  1. Missing an ignore_arg_check directive on loss_val in torchfort_train_c.
  2. Incorrect array dimensions for act in torchfort_rl_off_policy_predict_float_4d_2d leading to an ambiguous interface.