Update fused kernel error recommendation, and make error more verbose

EleutherAI · Mar 10, 2023 · d9b9bcb · d9b9bcb
1 parent 6edf241
commit d9b9bcb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/megatron/fused_kernels/__init__.py b/megatron/fused_kernels/__init__.py
@@ -33,11 +33,12 @@ def load_fused_kernels():
  try:
  import scaled_upper_triang_masked_softmax_cuda
  import scaled_masked_softmax_cuda
- except (ImportError, ModuleNotFoundError):
+ except (ImportError, ModuleNotFoundError) as e:
  print("\n")
+ print(e)
  print("=" * 100)
  print(
- f'ERROR: Fused kernels configured but not installed. Please run `python {str(srcpath / "setup.py")} install` to install them'
+ f'ERROR: Fused kernels configured but not installed. Please run `pip install {str(srcpath)}` to install them'
  )
  print("=" * 100)
  exit()